From 18ec2ba5623c5a9f9cbaf6a9563c94e704c32c81 Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Wed, 10 Jan 2024 16:27:38 +0200 Subject: [PATCH] tests(ai): add new field --- tests/ai/test_heuristics.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tests/ai/test_heuristics.py b/tests/ai/test_heuristics.py index 5375818..6490ffc 100644 --- a/tests/ai/test_heuristics.py +++ b/tests/ai/test_heuristics.py @@ -17,14 +17,31 @@ class TestHeuristics(unittest.TestCase): ] ) + self.field2 = np.array( + [ + [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], + [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], + [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], + [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], + [0, 0, 0, 0, 0, 0, 0, 0, 0, 1], + [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], + [1, 0, 0, 1, 1, 0, 0, 0, 0, 1], + [1, 1, 0, 1, 1, 0, 0, 0, 0, 1], + ] + ) + def test_aggregate_height(self) -> None: self.assertEqual(aggregate_height(self.field), 48) + self.assertEqual(aggregate_height(self.field2), 12) def test_complete_lines(self) -> None: self.assertEqual(complete_lines(self.field), 2) + self.assertEqual(complete_lines(self.field2), 0) def test_holes(self) -> None: self.assertEqual(count_holes(self.field), 2) + self.assertEqual(count_holes(self.field2), 0) def test_bumpiness(self) -> None: self.assertEqual(get_bumpiness(self.field), 6) + self.assertEqual(get_bumpiness(self.field2), 11)