From 32d4e69d583f1f1035dc2dc0edc4ddf9a40a08a7 Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Tue, 9 Jan 2024 20:30:38 +0200 Subject: [PATCH] refactor(ai): rename `holes` -> `count_holes` --- src/ai/heuristics/__init__.py | 4 ++-- src/ai/heuristics/holes.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 31cbfc3..b6e9509 100644 --- a/src/ai/heuristics/__init__.py +++ b/src/ai/heuristics/__init__.py @@ -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"] diff --git a/src/ai/heuristics/holes.py b/src/ai/heuristics/holes.py index f2a28aa..50bf441 100644 --- a/src/ai/heuristics/holes.py +++ b/src/ai/heuristics/holes.py @@ -1,7 +1,7 @@ import numpy as np -def holes( +def count_holes( field: np.ndarray[int, np.dtype[np.uint8]], ) -> int: """ diff --git a/tests/ai/test_heuristics.py b/tests/ai/test_heuristics.py index 268aecf..4eddfc0 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, 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)