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

refactor(ai): tests

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-05 17:42:46 +02:00
parent aca113f562
commit b762fe6774
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/ai/test_fitness.py
View File

@ -2,6 +2,7 @@ import unittest
import numpy as np
from ai.fitness.peaks import get_peaks_sum
from ai.fitness.transitions import get_row_transitions
class TestFitness(unittest.TestCase):
@ -29,3 +30,7 @@ class TestFitness(unittest.TestCase):
)
self.assertEqual(get_peaks_sum(field), 2)
def test_get_row_transistions(self):
answers = (8, 0, 2)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_row_transitions(field), answer)