Adding tests that bots certain hints for mines and safes are correct.

5 files changed, 66 insertions, 1 deletions

diff --git a/README.md b/README.md
index 4f0fd32..d7f52bc 100644
--- a/README.md
+++ b/README.md
@@ -30,4 +30,4 @@ Miinaharava ratkaisijalla
 `python3 miinaharava -c 10`
 
 ### Pelaa kentät tiedostosta
-`python3 miinaharava -f miinaharava/tests/data/beginner_3win.txt`
+`python3 miinaharava -f miinaharava/testdata/beginner_3win.txt`
diff --git a/tests/data/beginner_3win.txt b/testdata/beginner_3win.txt
index 6eeb1a4..6eeb1a4 100644
--- a/tests/data/beginner_3win.txt
+++ b/testdata/beginner_3win.txt
diff --git a/tests/data/dssp_win.txt b/testdata/dssp_win.txt
index 4a19dee..4a19dee 100644
--- a/tests/data/dssp_win.txt
+++ b/testdata/dssp_win.txt
diff --git a/testdata/not_solved.txt b/testdata/not_solved.txt
new file mode 100644
index 0000000..4a031c8
--- /dev/null
+++ b/testdata/not_solved.txt
@@ -0,0 +1,11 @@
+# Lautoja joissa dssp-ratkaisijalla tulee ongelmia
+
+.@.......
+@..@.....
+@.@......
+.........
+.........
+.........
+.........
+.........
+.........
diff --git a/tests/test_bot.py b/tests/test_bot.py
new file mode 100644
index 0000000..4dab148
--- /dev/null
+++ b/tests/test_bot.py
@@ -0,0 +1,54 @@
+""" tests/test_bot.py - Testaa botin toimintaa"""
+# pylint: disable = missing-class-docstring, too-few-public-methods, protected-access
+
+import unittest
+
+from board import Board, Tile
+from bots import DSSPBot, SimpleBot
+from tui import Action
+
+class TestBotClass(unittest.TestCase):
+    """ botin testit"""
+    def test_init(self):
+        """ olioden luominen onnistuu """
+        DSSPBot()
+        SimpleBot()
+
+    def correctly_marking(self, open_free=False, bot_type=DSSPBot):
+        """ Testaa onko miinat miinoja ja vapaat vapaita alkuun avatusta """
+        for _ in range(500):
+            brd = Board()
+            # jos ei aukea ylälaidasta otetaan seuraava
+            if not brd.guess(0,0):
+                continue
+            # vain varmat liikut
+            bot = bot_type(uncertain=False)
+
+            tested = set()
+            while True:
+                action, x, y = bot.hint(brd.get_view(), 0, 0)
+                if (x,y) in tested:
+                    break
+                tested.add((x,y))
+                if action == Action.SAFE:
+                    self.assertTrue( brd._Board__tiles[x][y] < Tile.MINE )
+                    if open_free:
+                        brd.guess(x,y)
+                if action == Action.MINE:
+                    self.assertTrue( brd._Board__tiles[x][y] == Tile.MINE )
+
+    def test_dssp_marks_correctly_with_open(self):
+        """ Testaa onko dssp:n miinat miinoja ja avaa vapaat """
+        self.correctly_marking(True, DSSPBot)
+
+    def test_simple_marks_correctly_with_open(self):
+        """ Testaa onko dssp:n miinat miinoja ja avaa vapaat """
+        self.correctly_marking(True, SimpleBot)
+
+    def test_dssp_marks_correctly(self):
+        """ Testaa onko dssp:n miinat miinoja ja vapaat vapaita """
+        self.correctly_marking(False, DSSPBot)
+
+    def test_simple_marks_correctly(self):
+        """ Testaa onko simple:n miinat miinoja ja vapaat vapaita """
+        self.correctly_marking(False, SimpleBot)