def test_partial_counts_in_rows(self):
threshold = 0.2
N = len(fps)
counts = array.array("i", [0]*N)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 1, 5, 9, 30)
expected = [0] * N
slow_counts(expected, fps, threshold, 1, 5, 9, 30)
self.assertSequenceEqual(counts, expected)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 20, 26, 30, 50)
slow_counts(expected, fps, threshold, 20, 26, 30, 50)
self.assertSequenceEqual(counts, expected)
def test_partial_counts_in_columns(self):
threshold = 0.6
N = len(fps)
counts = array.array("i", [0]*N)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 1, 5, 0, N)
expected = [0] * N
slow_counts(expected, fps, threshold, 1, 5, 0, N)
self.assertSequenceEqual(counts, expected)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold,
3, 33, 0, N)
slow_counts(expected, fps, threshold, 3, 33, 0, N)
self.assertSequenceEqual(counts, expected)
def test_partial_counts(self):
threshold = 0.2
N = len(fps)
counts = array.array("i", [0]*N)
expected = [0] * N
for i in range(0, N, 10):
for j in range(i, N, 8):
slow_counts(expected, fps, threshold, i, i+10, j, j+8)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold,
i, i+10, j, j+8)
self.assertSequenceEqual(counts, expected)
normal = search.count_tanimoto_hits_symmetric(fps, threshold)
self.assertSequenceEqual(normal, expected)
def test_partial_counts_in_columns(self):
threshold = 0.6
N = len(fps)
counts = array.array("i", [0] * N)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 1,
5, 0, N)
expected = [0] * N
slow_counts(expected, fps, threshold, 1, 5, 0, N)
self.assertSequenceEqual(counts, expected)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 3,
33, 0, N)
slow_counts(expected, fps, threshold, 3, 33, 0, N)
self.assertSequenceEqual(counts, expected)
def test_partial_counts(self):
threshold = 0.2
N = len(fps)
counts = array.array("i", [0] * N)
expected = [0] * N
for i in range(0, N, 10):
for j in range(i, N, 8):
slow_counts(expected, fps, threshold, i, i + 10, j, j + 8)
search.partial_count_tanimoto_hits_symmetric(
counts, fps, threshold, i, i + 10, j, j + 8)
self.assertSequenceEqual(counts, expected)
normal = search.count_tanimoto_hits_symmetric(fps, threshold)
self.assertSequenceEqual(normal, expected)
def test_partial_counts_in_rows(self):
threshold = 0.2
N = len(fps)
counts = array.array("i", [0] * N)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold, 1,
5, 9, 30)
expected = [0] * N
slow_counts(expected, fps, threshold, 1, 5, 9, 30)
self.assertSequenceEqual(counts, expected)
search.partial_count_tanimoto_hits_symmetric(counts, fps, threshold,
20, 26, 30, 50)
slow_counts(expected, fps, threshold, 20, 26, 30, 50)
self.assertSequenceEqual(counts, expected)
