py/obj: Add support for __float__ and __complex__ functions.

This commit is contained in:
Andrew Leech
2022-07-04 17:35:46 +10:00
committed by Damien George
parent fa15aed0f7
commit 1e87b56219
9 changed files with 130 additions and 3 deletions

View File

@@ -0,0 +1,40 @@
# test __complex__ function support
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(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")