summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_complex.py
diff options
context:
space:
mode:
authorMark Dickinson <mdickinson@enthought.com>2021-08-23 08:15:49 (GMT)
committerGitHub <noreply@github.com>2021-08-23 08:15:49 (GMT)
commit6082bb5addab93755ab6e2bd2ed6021b391e10d1 (patch)
treeef109d492632072d919a7ebeb2ce89342a2babdb /Lib/test/test_complex.py
parenteec340ea3af27887fcaac4029ebdee99f3713bff (diff)
downloadcpython-6082bb5addab93755ab6e2bd2ed6021b391e10d1.zip
cpython-6082bb5addab93755ab6e2bd2ed6021b391e10d1.tar.gz
cpython-6082bb5addab93755ab6e2bd2ed6021b391e10d1.tar.bz2
bpo-24234: implement complex.__complex__ (GH-27887)
Co-authored-by: Dong-hee Na <donghee.na92@gmail.com>
Diffstat (limited to 'Lib/test/test_complex.py')
-rw-r--r--Lib/test/test_complex.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/Lib/test/test_complex.py b/Lib/test/test_complex.py
index abd7e39..1cd025e 100644
--- a/Lib/test/test_complex.py
+++ b/Lib/test/test_complex.py
@@ -499,6 +499,18 @@ class ComplexTest(unittest.TestCase):
self.assertEqual(complex(complex1(1j)), 2j)
self.assertRaises(TypeError, complex, complex2(1j))
+ def test___complex__(self):
+ z = 3 + 4j
+ self.assertEqual(z.__complex__(), z)
+ self.assertEqual(type(z.__complex__()), complex)
+
+ class complex_subclass(complex):
+ pass
+
+ z = complex_subclass(3 + 4j)
+ self.assertEqual(z.__complex__(), 3 + 4j)
+ self.assertEqual(type(z.__complex__()), complex)
+
@support.requires_IEEE_754
def test_constructor_special_numbers(self):
class complex2(complex):