From 391c0c70dd4a16846fcf29344aa0d35cf2cc9e16 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Wed, 21 Jun 2023 12:24:59 -0700 Subject: [PATCH] Backport PR #53774: COMPAT: Remove np.compat --- pandas/tests/libs/test_hashtable.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pandas/tests/libs/test_hashtable.py b/pandas/tests/libs/test_hashtable.py index 4a5d5e0c85a09..60af595018a54 100644 --- a/pandas/tests/libs/test_hashtable.py +++ b/pandas/tests/libs/test_hashtable.py @@ -461,7 +461,7 @@ def test_get_labels_groupby_for_Int64(writable): def test_tracemalloc_works_for_StringHashTable(): N = 1000 - keys = np.arange(N).astype(np.compat.unicode).astype(np.object_) + keys = np.arange(N).astype(np.str_).astype(np.object_) with activated_tracemalloc(): table = ht.StringHashTable() table.map_locations(keys) @@ -484,7 +484,7 @@ def test_tracemalloc_for_empty_StringHashTable(): @pytest.mark.parametrize("N", range(1, 110)) def test_no_reallocation_StringHashTable(N): - keys = np.arange(N).astype(np.compat.unicode).astype(np.object_) + keys = np.arange(N).astype(np.str_).astype(np.object_) preallocated_table = ht.StringHashTable(N) n_buckets_start = preallocated_table.get_state()["n_buckets"] preallocated_table.map_locations(keys)