refactor(ai): add Optional

fix

fix
This commit is contained in:
Kristofers Solo
2024-01-05 18:04:17 +02:00
parent 4ab9b4bb2a
commit 0d4ab8aab7
4 changed files with 30 additions and 15 deletions

View File

@@ -32,7 +32,7 @@ class TestFitness(unittest.TestCase):
def test_get_peaks_sum(self) -> None:
answers: tuple[int] = (11, 0, 2)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_peaks_sum(field), answer)
self.assertEqual(get_peaks_sum(None, field), answer)
def test_get_row_transistions(self):
answers = (8, 0, 2)
@@ -45,6 +45,6 @@ class TestFitness(unittest.TestCase):
self.assertEqual(get_col_transition(field), answer)
def test_get_bumpiness(self):
answers = (8, 0, 2)
answers = (5, 0, 4)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_bumpiness(field), answer)
self.assertEqual(get_bumpiness(None, field), answer)