Add two more similar tests.

jwg4 · Nov 15, 2023 · d214e73 · d214e73
1 parent 584448d
commit d214e73
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_solution_count.py b/tests/test_solution_count.py
@@ -8,3 +8,15 @@ def test_solution_count():
     data = np.array([[1, 0, 0], [0, 1, 0], [0, 1, 1], [0, 0, 1]], dtype=DTYPE_FOR_ARRAY)
     result = get_solution_count(data)
     assert result == 2
+
+
+def test_solution_count_single_solution():
+    data = np.array([[1, 0, 0], [0, 1, 0], [0, 1, 1]], dtype=DTYPE_FOR_ARRAY)
+    result = get_solution_count(data)
+    assert result == 1
+
+
+def test_solution_count_no_solutions():
+    data = np.array([[1, 0, 0, 0], [0, 1, 0, 0], [0, 1, 1, 0], [0, 0, 1, 0]], dtype=DTYPE_FOR_ARRAY)
+    result = get_solution_count(data)
+    assert result == 0