mirror of
https://github.com/kristoferssolo/Tetris.git
synced 2025-10-21 20:00:35 +00:00
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
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):
|
|
def setUp(self) -> None:
|
|
self.fields: tuple[np.ndarray] = (
|
|
np.array(
|
|
[
|
|
[0, 1, 0, 0, 1],
|
|
[1, 0, 0, 1, 0],
|
|
[0, 1, 1, 0, 0],
|
|
]
|
|
),
|
|
np.zeros((3, 5)),
|
|
np.array(
|
|
[
|
|
[0, 0, 0, 0, 0],
|
|
[0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0],
|
|
]
|
|
),
|
|
)
|
|
|
|
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)
|
|
|
|
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)
|