tests/float/cmath_dunder.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

# test that cmath functions support user classes with __float__ and __complex__

try:
    import cmath
except ImportError:
    print("SKIP")
    raise SystemExit


class TestFloat:
    def __float__(self):
        return 1.0


class TestComplex:
    def __complex__(self):
        return complex(10, 1)


for clas in TestFloat, TestComplex:
    print("%.5g" % cmath.phase(clas()))