From e4720c779da8bc27df731373e718ce260efce7da Mon Sep 17 00:00:00 2001
From: Kristofers Solo <kristians.cagulis@pm.me>
Date: Tue, 9 Jan 2024 20:21:14 +0200
Subject: [PATCH] tests(ai): set setUp for heuristics

---
 tests/ai/test_heuristics.py | 17 +++++++++++++++++
 tests/test_blank.py         |  6 ------
 2 files changed, 17 insertions(+), 6 deletions(-)
 create mode 100644 tests/ai/test_heuristics.py
 delete mode 100644 tests/test_blank.py

diff --git a/tests/ai/test_heuristics.py b/tests/ai/test_heuristics.py
new file mode 100644
index 0000000..d58d483
--- /dev/null
+++ b/tests/ai/test_heuristics.py
@@ -0,0 +1,17 @@
+import unittest
+
+import numpy as np
+
+
+class TestHeuristics(unittest.TestCase):
+    def setUp(self) -> None:
+        self.field = np.array(
+            [
+                [0, 0, 0, 0, 1, 1, 0, 0, 0, 0],
+                [0, 1, 1, 1, 1, 1, 1, 0, 0, 1],
+                [0, 1, 1, 0, 1, 1, 1, 1, 1, 1],
+                [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
+                [1, 1, 1, 0, 1, 1, 1, 1, 1, 1],
+                [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
+            ]
+        )
diff --git a/tests/test_blank.py b/tests/test_blank.py
deleted file mode 100644
index a59017a..0000000
--- a/tests/test_blank.py
+++ /dev/null
@@ -1,6 +0,0 @@
-import unittest
-
-
-class TestBlank(unittest.TestCase):
-    def test(self) -> None:
-        pass