@@ -30,9 +30,11 @@ def get_allocated_khash_memory():
30
30
"table_type, dtype" ,
31
31
[
32
32
(ht .PyObjectHashTable , np .object_ ),
33
+ (ht .Complex128HashTable , np .complex128 ),
33
34
(ht .Int64HashTable , np .int64 ),
34
35
(ht .UInt64HashTable , np .uint64 ),
35
36
(ht .Float64HashTable , np .float64 ),
37
+ (ht .Complex64HashTable , np .complex64 ),
36
38
(ht .Int32HashTable , np .int32 ),
37
39
(ht .UInt32HashTable , np .uint32 ),
38
40
(ht .Float32HashTable , np .float32 ),
@@ -182,6 +184,8 @@ def test_tracemalloc_for_empty_StringHashTable():
182
184
[
183
185
(ht .Float64HashTable , np .float64 ),
184
186
(ht .Float32HashTable , np .float32 ),
187
+ (ht .Complex128HashTable , np .complex128 ),
188
+ (ht .Complex64HashTable , np .complex64 ),
185
189
],
186
190
)
187
191
class TestHashTableWithNans :
@@ -233,9 +237,11 @@ def get_ht_function(fun_name, type_suffix):
233
237
"dtype, type_suffix" ,
234
238
[
235
239
(np .object_ , "object" ),
240
+ (np .complex128 , "complex128" ),
236
241
(np .int64 , "int64" ),
237
242
(np .uint64 , "uint64" ),
238
243
(np .float64 , "float64" ),
244
+ (np .complex64 , "complex64" ),
239
245
(np .int32 , "int32" ),
240
246
(np .uint32 , "uint32" ),
241
247
(np .float32 , "float32" ),
@@ -304,6 +310,8 @@ def test_mode(self, dtype, type_suffix):
304
310
[
305
311
(np .float64 , "float64" ),
306
312
(np .float32 , "float32" ),
313
+ (np .complex128 , "complex128" ),
314
+ (np .complex64 , "complex64" ),
307
315
],
308
316
)
309
317
class TestHelpFunctionsWithNans :
0 commit comments