kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2025-10-21 20:00:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests(ai): add new field

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-10 16:27:38 +02:00
parent 20f9b15001
commit 18ec2ba562
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/ai/test_heuristics.py
View File

@ -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)