import pytest import random from numpy.core._multiarray_tests import identityhash_tester @pytest.mark.parametrize("key_length", [1, 3, 6]) @pytest.mark.parametrize("length", [1, 16, 2000]) def test_identity_hashtable(key_length, length): # use a 30 object pool for everything (duplicates will happen) pool = [object() for i in range(20)] keys_vals = [] for i in range(length): keys = tuple(random.choices(pool, k=key_length)) keys_vals.append((keys, random.choice(pool))) dictionary = dict(keys_vals) # add a random item at the end: keys_vals.append(random.choice(keys_vals)) # the expected one could be different with duplicates: expected = dictionary[keys_vals[-1][0]] res = identityhash_tester(key_length, keys_vals, replace=True) assert res is expected # check that ensuring one duplicate definitely raises: keys_vals.insert(0, keys_vals[-2]) with pytest.raises(RuntimeError): identityhash_tester(key_length, keys_vals)