diff options
| author | Ganesh Kathiresan <ganesh3597@gmail.com> | 2021-05-30 23:19:27 +0530 |
|---|---|---|
| committer | Ganesh Kathiresan <ganesh3597@gmail.com> | 2021-05-30 23:19:27 +0530 |
| commit | f59e22561e02a185e57d67d07b423fd429ea35bb (patch) | |
| tree | 184a038c1d8534b66ff5012d574312df8e7335da /numpy | |
| parent | 87ae1738b20ddf7e0644caf30e4c08b4bf83e9ef (diff) | |
| download | numpy-f59e22561e02a185e57d67d07b423fd429ea35bb.tar.gz | |
BUG, TST: Determine complex types using typecodes
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/ma/tests/test_core.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py index a3a109a1c..15ed26470 100644 --- a/numpy/ma/tests/test_core.py +++ b/numpy/ma/tests/test_core.py @@ -2854,7 +2854,7 @@ class TestMaskedArrayInPlaceArithmetics: 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 @@ class TestMaskedArrayInPlaceArithmetics: 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") |