tests/float/complex_dunder.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

# test __complex__ function support


class TestFloat:
    def __float__(self):
        return 1.0


class TestComplex:
    def __complex__(self):
        return 1j + 10


class TestStrComplex:
    def __complex__(self):
        return "a"


class TestNonComplex:
    def __complex__(self):
        return 6


class Test:
    pass


print(complex(TestFloat()))
print(complex(TestComplex()))

try:
    print(complex(TestStrComplex()))
except TypeError:
    print("TypeError")


try:
    print(complex(TestNonComplex()))
except TypeError:
    print("TypeError")


try:
    print(complex(Test()))
except TypeError:
    print("TypeError")