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

delete ai

Browse Source
This commit is contained in:
Kristofers Solo
2024-01-07 15:29:55 +02:00
parent 77cd153b10
commit 06f962f5c7
23 changed files with 2 additions and 951 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
tests/ai/test_fitness.py
View File

@@ -1,70 +0,0 @@
import unittest
import numpy as np
from ai.fitness.bumpiness import get_bumpiness
from ai.fitness.holes import holes
from ai.fitness.peaks import get_peaks_sum
from ai.fitness.transitions import (
get_col_transition,
get_row_transition,
)
from ai.fitness.wells import get_wells
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=field), answer)
def test_get_row_transistions(self) -> None:
answers = (8, 0, 2)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_row_transition(field), answer)
def test_get_col_transistions(self) -> None:
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 = (5, 0, 4)
for field, answer in zip(self.fields, answers):
self.assertEqual(get_bumpiness(field=field), answer)
def test_get_holes(self) -> None:
answers = (
np.array([1, 1, 0, 1, 2]),
np.array([0, 0, 0, 0, 0]),
np.array([0, 1, 0, 0, 0]),
)
for field, answer in zip(self.fields, answers):
self.assertTrue(np.array_equal(holes(field), answer))
def test_get_wells(self) -> None:
answers = (
np.array([1, 0, 2, 1, 0]),
np.array([0, 0, 0, 0, 0]),
np.array([2, 0, 2, 0, 0]),
)
for field, answer in zip(self.fields, answers):
self.assertTrue(np.array_equal(get_wells(field=field), answer))

33
tests/ai/test_moves.py
View File

@@ -1,33 +0,0 @@
import unittest
import numpy as np
from ai.moves.bumpiness import bumpiness
from ai.moves.height import aggregate_height
from ai.moves.holes import holes
from ai.moves.lines import complete_lines
class TestFitness(unittest.TestCase):
def setUp(self) -> None:
self.field = np.array(
[
[0, 0, 0, 0, 1, 1, 0, 0, 0, 0],
[0, 1, 1, 1, 1, 1, 1, 0, 0, 1],
[0, 1, 1, 0, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 0, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
]
)
def test_aggregate_height(self) -> None:
self.assertEqual(aggregate_height(self.field), 48)
def test_complete_lines(self) -> None:
self.assertEqual(complete_lines(self.field), 2)
def test_holes(self) -> None:
self.assertEqual(holes(self.field), 2)
def test_bumpiness(self) -> None:
self.assertEqual(bumpiness(self.field), 6)