refactor(ai): rename bumpiness -> get_bumpiness

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

View File

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

View File

@ -1,7 +1,7 @@
import numpy as np
def bumpiness(
def get_bumpiness(
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, count_holes
from ai.heuristics import aggregate_height, complete_lines, count_holes, get_bumpiness
class TestHeuristics(unittest.TestCase):
@ -27,4 +27,4 @@ class TestHeuristics(unittest.TestCase):
self.assertEqual(count_holes(self.field), 2)
def test_bumpiness(self) -> None:
self.assertEqual(bumpiness(self.field), 6)
self.assertEqual(get_bumpiness(self.field), 6)