Skip to content

Commit 873f2a2

Browse files
committed
activating tests for complex hashtables
1 parent 2d5d0f9 commit 873f2a2

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pandas/tests/libs/test_hashtable.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,11 @@ def get_allocated_khash_memory():
3030
"table_type, dtype",
3131
[
3232
(ht.PyObjectHashTable, np.object_),
33+
(ht.Complex128HashTable, np.complex128),
3334
(ht.Int64HashTable, np.int64),
3435
(ht.UInt64HashTable, np.uint64),
3536
(ht.Float64HashTable, np.float64),
37+
(ht.Complex64HashTable, np.complex64),
3638
(ht.Int32HashTable, np.int32),
3739
(ht.UInt32HashTable, np.uint32),
3840
(ht.Float32HashTable, np.float32),
@@ -182,6 +184,8 @@ def test_tracemalloc_for_empty_StringHashTable():
182184
[
183185
(ht.Float64HashTable, np.float64),
184186
(ht.Float32HashTable, np.float32),
187+
(ht.Complex128HashTable, np.complex128),
188+
(ht.Complex64HashTable, np.complex64),
185189
],
186190
)
187191
class TestHashTableWithNans:
@@ -233,9 +237,11 @@ def get_ht_function(fun_name, type_suffix):
233237
"dtype, type_suffix",
234238
[
235239
(np.object_, "object"),
240+
(np.complex128, "complex128"),
236241
(np.int64, "int64"),
237242
(np.uint64, "uint64"),
238243
(np.float64, "float64"),
244+
(np.complex64, "complex64"),
239245
(np.int32, "int32"),
240246
(np.uint32, "uint32"),
241247
(np.float32, "float32"),
@@ -304,6 +310,8 @@ def test_mode(self, dtype, type_suffix):
304310
[
305311
(np.float64, "float64"),
306312
(np.float32, "float32"),
313+
(np.complex128, "complex128"),
314+
(np.complex64, "complex64"),
307315
],
308316
)
309317
class TestHelpFunctionsWithNans:

0 commit comments

Comments
 (0)