From 7839406da38638ccd1e3b183137b8cb0cd0619bb Mon Sep 17 00:00:00 2001 From: "Lumberbot (aka Jack)" <39504233+meeseeksmachine@users.noreply.github.com> Date: Wed, 21 Jun 2023 13:39:30 -0700 Subject: [PATCH] Backport PR #53774 on branch 2.0.x (COMPAT: Remove np.compat) (#53776) Backport PR #53774: COMPAT: Remove np.compat Co-authored-by: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> --- 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)