more tests

outbrain · Aug 16, 2024 · 695e7f9 · 695e7f9
1 parent fadc249
commit 695e7f9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/cov_heu_test.py b/tests/cov_heu_test.py
@@ -42,3 +42,15 @@ def test_all_same_pairs(self):
         array2 = np.array([2, 2, 2, 2, 2])
         result = max_pair_coverage(array1, array2)
         self.assertEqual(result, 1.0)
+
+    def test_high_collision_potential(self):
+        array1 = np.array([1] * 1000)
+        array2 = np.array([2] * 1000)
+        result = max_pair_coverage(array1, array2)
+        self.assertEqual(result, 1.0)
+
+    def test_very_large_arrays(self):
+        array1 = np.random.randint(0, 1000, size=1000000)
+        array2 = np.random.randint(0, 1000, size=1000000)
+        result = max_pair_coverage(array1, array2)
+        self.assertTrue(0 <= result <= 1)