summaryrefslogtreecommitdiff
path: root/tests/test_board.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_board.py')
-rw-r--r--tests/test_board.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_board.py b/tests/test_board.py
index 904f73a..7b0d1fd 100644
--- a/tests/test_board.py
+++ b/tests/test_board.py
@@ -108,6 +108,6 @@ class TestBoardClass(unittest.TestCase):
def test_get_level(self):
""" Testataan että nykyinen vaikeustaso palautuu oikein """
b = Board(level=Level.INTERMEDIATE)
- self.assertEqual(b.get_level(), Level.INTERMEDIATE)
- b = Board(level=Level.INTERMEDIATE, width=25)
- self.assertEqual(b.get_level(), None)
+ self.assertIn(LevelSpecs[Level.INTERMEDIATE][3], b.get_level_name())
+ b = Board(level=Level.INTERMEDIATE, width=25, bombs=2)
+ self.assertIn("Mukautettu", b.get_level_name())