From 85254c572474648d53590f549c75fe248a8ac44d Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Fri, 5 Jan 2024 17:14:14 +0200 Subject: [PATCH] feat(ai): add `get_peaks` unittests --- tests/ai/test_fitness.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 tests/ai/test_fitness.py diff --git a/tests/ai/test_fitness.py b/tests/ai/test_fitness.py new file mode 100644 index 0000000..7241c23 --- /dev/null +++ b/tests/ai/test_fitness.py @@ -0,0 +1,31 @@ +import unittest + +import numpy as np +from ai.fitness.peaks import get_peaks + + +class TestFitness(unittest.TestCase): + def test_get_peaks(self) -> None: + field = np.array( + [ + [0, 1, 0, 0, 1], + [1, 0, 0, 1, 0], + [0, 1, 1, 0, 0], + ] + ) + self.assertEqual(get_peaks(field), 11) + + def test_get_peaks_zeros(self) -> None: + field = np.zeros((3, 5)) + self.assertEqual(get_peaks(field), 0) + + def test_single_peak(self): + field = np.array( + [ + [0, 0, 0, 0, 0], + [0, 1, 0, 0, 0], + [0, 0, 0, 0, 0], + ] + ) + + self.assertEqual(get_peaks(field), 2)