refactor(ai): rename holes -> count_holes

This commit is contained in:
Kristofers Solo 2024-01-09 20:30:38 +02:00
parent 5c0d07ba15
commit 32d4e69d58
3 changed files with 5 additions and 5 deletions

View File

@ -1,6 +1,6 @@
from .bumpiness import bumpiness
from .height import aggregate_height
from .holes import holes
from .holes import count_holes
from .lines import complete_lines
__all__ = ["aggregate_height", "bumpiness", "complete_lines", "holes"]
__all__ = ["aggregate_height", "bumpiness", "complete_lines", "count_holes"]

View File

@ -1,7 +1,7 @@
import numpy as np
def holes(
def count_holes(
field: np.ndarray[int, np.dtype[np.uint8]],
) -> int:
"""

View File

@ -1,7 +1,7 @@
import unittest
import numpy as np
from ai.heuristics import aggregate_height, bumpiness, complete_lines, holes
from ai.heuristics import aggregate_height, bumpiness, complete_lines, count_holes
class TestHeuristics(unittest.TestCase):
@ -24,7 +24,7 @@ class TestHeuristics(unittest.TestCase):
self.assertEqual(complete_lines(self.field), 2)
def test_holes(self) -> None:
self.assertEqual(holes(self.field), 2)
self.assertEqual(count_holes(self.field), 2)
def test_bumpiness(self) -> None:
self.assertEqual(bumpiness(self.field), 6)