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

feat(ai): add get_col_transition

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-05 17:50:36 +02:00
parent dec7c4d0e0
commit f673a9f850
2 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/ai/fitness/transitions.py
View File

@ -1,9 +1,9 @@
import numpy as np
from .peaks import get_peaks_max
from .peaks import get_peaks, get_peaks_max
def get_row_transitions(field: np.ndarray) -> int:
def get_row_transition(field: np.ndarray) -> int:
highest_peak = get_peaks_max(field)
rows_to_check = slice(int(field.shape[0] - highest_peak), field.shape[0])
@ -12,5 +12,17 @@ def get_row_transitions(field: np.ndarray) -> int:
return int(transitions)
def get_col_transitions(field: np.ndarray) -> int:
pass
def get_col_transition(field: np.ndarray) -> int:
transitions_sum = 0
peaks = get_peaks(field)
for col in range(field.shape[1]):
if peaks[col] <= 1:
continue
col_values = field[int(field.shape[0] - peaks[col]) : field.shape[0], col]
transitions = np.sum(col_values[:-1] != col_values[1:])
transitions_sum += transitions
return transitions_sum

13
tests/ai/test_fitness.py
View File

@ -2,7 +2,11 @@ import unittest
import numpy as np
from ai.fitness.peaks import get_peaks_sum
from ai.fitness.transitions import get_row_transitions
from ai.fitness.transitions import (
get_col_transition,
get_col_transitions2,
get_row_transition,
)
class TestFitness(unittest.TestCase):
@ -33,4 +37,9 @@ class TestFitness(unittest.TestCase):
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)
self.assertEqual(get_row_transition(field), answer)
def test_get_col_transistions2(self):
answers = (5, 0, 1)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_col_transition(field), answer)