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)