Skip to content

Commit dd642c1

Browse files
committed
feedback: use lib.is_complex
1 parent 640afc1 commit dd642c1

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

pandas/_libs/testing.pyx

+2-12
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ from numpy cimport import_array
88
import_array()
99

1010
from pandas._libs.util cimport is_array
11+
from pandas._libs.lib import is_complex
1112

1213
from pandas.core.dtypes.common import is_dtype_equal
1314
from pandas.core.dtypes.missing import array_equivalent, isna
@@ -32,21 +33,10 @@ cdef NUMERIC_TYPES = (
3233
)
3334

3435

35-
cdef COMPLEX_NUMERIC_TYPES = (
36-
complex,
37-
np.complex64,
38-
np.complex128,
39-
)
40-
41-
4236
cdef bint is_comparable_as_number(obj):
4337
return isinstance(obj, NUMERIC_TYPES)
4438

4539

46-
cdef bint is_comparable_as_complex_number(obj):
47-
return isinstance(obj, COMPLEX_NUMERIC_TYPES)
48-
49-
5040
cdef bint isiterable(obj):
5141
return hasattr(obj, '__iter__')
5242

@@ -222,7 +212,7 @@ cpdef assert_almost_equal(a, b,
222212
f"with rtol={rtol}, atol={atol}")
223213
return True
224214

225-
if is_comparable_as_complex_number(a) and is_comparable_as_complex_number(b):
215+
if is_complex(a) and is_complex(b):
226216
if array_equivalent(a, b, strict_nan=True):
227217
# inf comparison
228218
return True

0 commit comments

Comments
 (0)