Merge pull request #21 from cbhyphen/patch-1

hiive · Oct 7, 2023 · 48815b7 · 48815b7
2 parents efbc7ce + 9085831
commit 48815b7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_fitness.py b/tests/test_fitness.py
@@ -191,10 +191,10 @@ def test_queens():
     @staticmethod
     def test_max_k_color():
         """Test MaxKColor fitness function"""
-        edges = [(0, 1), (0, 2), (0, 4), (1, 3), (2, 0), (2, 3), (3, 4)]
+        edges = [(0, 1), (0, 2), (0, 4), (1, 3), (2, 0), (2, 3), (3, 4), (0, 5)]
 
-        state = np.array([0, 1, 0, 1, 1])
-        assert MaxKColor(edges).evaluate(state) == 3
+        state = np.array([0, 1, 0, 1, 1, 1])
+        assert MaxKColor(edges).evaluate(state) == 4
 
     @staticmethod
     def test_custom_fitness():