BUG, TST: Determine complex types using typecodes

numpy · May 30, 2021 · f59e225 · f59e225
1 parent 87ae173
commit f59e225
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
@@ -2854,7 +2854,7 @@ def test_inplace_multiplication_array_type(self):
     def test_inplace_floor_division_scalar_type(self):
         # Test of inplace division
         # Check for TypeError in case of unsupported types
-        unsupported = {np.complex64, np.complex128, np.complex256}
+        unsupported = {np.dtype(t).type for t in np.typecodes["Complex"]}
         for t in self.othertypes:
             with warnings.catch_warnings(record=True) as w:
                 warnings.filterwarnings("always")
@@ -2876,7 +2876,7 @@ def test_inplace_floor_division_scalar_type(self):
     def test_inplace_floor_division_array_type(self):
         # Test of inplace division
         # Check for TypeError in case of unsupported types
-        unsupported = {np.complex64, np.complex128, np.complex256}
+        unsupported = {np.dtype(t).type for t in np.typecodes["Complex"]}
         for t in self.othertypes:
             with warnings.catch_warnings(record=True) as w:
                 warnings.filterwarnings("always")