From 9c4b6977229139255d7d6f49ab9021293798d2fe Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Tue, 9 Jan 2024 20:32:42 +0200 Subject: [PATCH] refactor(ai): rename `bumpiness` -> `get_bumpiness` --- src/ai/heuristics/__init__.py | 4 ++-- src/ai/heuristics/bumpiness.py | 2 +- tests/ai/test_heuristics.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/ai/heuristics/__init__.py b/src/ai/heuristics/__init__.py index b6e9509..30920fb 100644 --- a/src/ai/heuristics/__init__.py +++ b/src/ai/heuristics/__init__.py @@ -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"] diff --git a/src/ai/heuristics/bumpiness.py b/src/ai/heuristics/bumpiness.py index 17c57ae..3632cc4 100644 --- a/src/ai/heuristics/bumpiness.py +++ b/src/ai/heuristics/bumpiness.py @@ -1,7 +1,7 @@ import numpy as np -def bumpiness( +def get_bumpiness( field: np.ndarray[int, np.dtype[np.uint8]], ) -> int: """ diff --git a/tests/ai/test_heuristics.py b/tests/ai/test_heuristics.py index 4eddfc0..5375818 100644 --- a/tests/ai/test_heuristics.py +++ b/tests/ai/test_heuristics.py @@ -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)