def test_winnow_density(self):
in_range = True
for win_size in range(2, 20):
expected_density = 2 / (win_size + 1) * 10000
rand_hashes = np.random.randint(-9223372036854775808,
9223372036854775807,
10000,
dtype=np.int64)
density = len(_winnow(rand_hashes, win_size))
if not (density > expected_density - 200
and density < expected_density + 200):
in_range = False
assert in_range
def test_winnow_inf(self):
winnow_25 = np.array([0, 8, 19])
out_25 = _winnow(self.test_arr, 25)
assert np.array_equal(out_25, winnow_25)
def test_winnow_3(self):
winnow_3 = np.array([0, 2, 5, 8, 11, 12, 14, 16, 19])
out_3 = _winnow(self.test_arr, 3)
assert np.array_equal(out_3, winnow_3)
def test_winnow_2(self):
winnow_2 = np.array([0, 2, 4, 5, 6, 8, 9, 11, 12, 14, 16, 18, 19])
out_2 = _winnow(self.test_arr, 2)
assert np.array_equal(out_2, winnow_2)
def test_winnow_1(self):
winnow_1 = np.array(np.arange(20))
out_1 = _winnow(self.test_arr, 1)
assert np.array_equal(out_1, winnow_1)
def test_winnow_empty(self):
winnow_empty = np.array([])
out_empty = _winnow(np.array([]), 3)
assert np.array_equal(out_empty, winnow_empty)