kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2026-03-22 00:36:20 +00:00
Code Issues Packages Projects Releases 4 Wiki Activity

feat(ai): add get_wells

Browse Source
This commit is contained in:
Kristofers Solo
2024-01-05 19:06:48 +02:00
parent 01a84bc201
commit 7fd44f2834
3 changed files with 47 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/ai/test_fitness.py
View File

@@ -8,6 +8,7 @@ from ai.fitness.transitions import (
get_col_transition,
get_row_transition,
)
from ai.fitness.wells import get_wells
class TestFitness(unittest.TestCase):
@@ -58,3 +59,12 @@ class TestFitness(unittest.TestCase):
)
for field, answer in zip(self.fields, answers):
self.assertTrue(np.array_equal(get_holes(field), answer))
def test_get_wells(self):
answers = (
np.array([1, 0, 2, 1, 0]),
np.array([0, 0, 0, 0, 0]),
np.array([2, 0, 2, 0, 0]),
)
for field, answer in zip(self.fields, answers):
self.assertTrue(np.array_equal(get_wells(field=field), answer))