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_bumpiness

Browse Source
This commit is contained in:
Kristofers Solo
2024-01-05 17:59:08 +02:00
parent f673a9f850
commit 4ab9b4bb2a
3 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/ai/test_fitness.py
View File

@@ -1,10 +1,10 @@
import unittest
import numpy as np
from ai.fitness.bumpiness import get_bumpiness
from ai.fitness.peaks import get_peaks_sum
from ai.fitness.transitions import (
get_col_transition,
get_col_transitions2,
get_row_transition,
)
@@ -39,7 +39,12 @@ class TestFitness(unittest.TestCase):
for field, answer in zip(self.fields, answers):
self.assertEqual(get_row_transition(field), answer)
def test_get_col_transistions2(self):
def test_get_col_transistions(self):
answers = (5, 0, 1)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_col_transition(field), answer)
def test_get_bumpiness(self):
answers = (8, 0, 2)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_bumpiness(field), answer)