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: add additional field

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-14 19:53:55 +02:00
parent ea97c0f384
commit 757faf0f93
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/ai/test_heuristics.py
View File

@ -29,19 +29,37 @@ class TestHeuristics(unittest.TestCase):
[1, 1, 0, 1, 1, 0, 0, 0, 0, 1], [1, 1, 0, 1, 1, 0, 0, 0, 0, 1],
] ]
) )
self.field3 = 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, 0],
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 1, 1, 0, 0, 0, 0, 0, 0, 1],
[1, 1, 1, 0, 0, 0, 0, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 0, 0, 1, 1],
]
)
def test_aggregate_height(self) -> None: def test_aggregate_height(self) -> None:
self.assertEqual(aggregate_height(self.field), 48) self.assertEqual(aggregate_height(self.field), 48)
self.assertEqual(aggregate_height(self.field2), 12) self.assertEqual(aggregate_height(self.field2), 12)
self.assertEqual(aggregate_height(self.field3), 30)
def test_complete_lines(self) -> None: def test_complete_lines(self) -> None:
self.assertEqual(complete_lines(self.field), 2) self.assertEqual(complete_lines(self.field), 2)
self.assertEqual(complete_lines(self.field2), 0) self.assertEqual(complete_lines(self.field2), 0)
self.assertEqual(complete_lines(self.field3), 1)
def test_holes(self) -> None: def test_holes(self) -> None:
self.assertEqual(count_holes(self.field), 2) self.assertEqual(count_holes(self.field), 2)
self.assertEqual(count_holes(self.field2), 0) self.assertEqual(count_holes(self.field2), 0)
self.assertEqual(count_holes(self.field3), 2)
def test_bumpiness(self) -> None: def test_bumpiness(self) -> None:
self.assertEqual(get_bumpiness(self.field), 6) self.assertEqual(get_bumpiness(self.field), 6)
self.assertEqual(get_bumpiness(self.field2), 11) self.assertEqual(get_bumpiness(self.field2), 11)
self.assertEqual(get_bumpiness(self.field3), 7)