pythongh-109802: Increase test coverage for complexobject.c

* _Py_c_pow: L134 (this case goes to c_powi() due to L523), L139 // line numbers wrt to 54fbfa8
skirpichev · Sep 24, 2023 · facb60b · facb60b
1 parent 51863b7
commit facb60b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/Lib/test/test_complex.py b/Lib/test/test_complex.py
@@ -212,7 +212,9 @@ def test_divmod_zero_division(self):
     def test_pow(self):
         self.assertAlmostEqual(pow(1+1j, 0+0j), 1.0)
         self.assertAlmostEqual(pow(0+0j, 2+0j), 0.0)
+        self.assertAlmostEqual(pow(0+0j, 2000+0j), 0.0)
         self.assertRaises(ZeroDivisionError, pow, 0+0j, 1j)
+        self.assertRaises(ZeroDivisionError, pow, 0+0j, -1000)
         self.assertAlmostEqual(pow(1j, -1), 1/1j)
         self.assertAlmostEqual(pow(1j, 200), 1)
         self.assertRaises(ValueError, pow, 1+1j, 1+1j, 1+1j)

diff --git a/Objects/complexobject.c b/Objects/complexobject.c
@@ -131,11 +131,8 @@ _Py_c_pow(Py_complex a, Py_complex b)
 {
     Py_complex r;
     double vabs,len,at,phase;
-    if (b.real == 0. && b.imag == 0.) {
-        r.real = 1.;
-        r.imag = 0.;
-    }
-    else if (a.real == 0. && a.imag == 0.) {
+    assert(!(b.real == 0. && b.imag == 0.));
+    if (a.real == 0. && a.imag == 0.) {
         if (b.imag != 0. || b.real < 0.)
             errno = EDOM;
         r.real = 0.;