Add test cases for utils module

Wiston999 · May 5, 2017 · e74cd7e · e74cd7e
1 parent da56d59
commit e74cd7e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -4,6 +4,7 @@
 import unittest
 from rubik_solver.Solver import Beginner, CFOP, Kociemba
 
+class MockSolver(object): pass
 
 class TestUtils(unittest.TestCase):
     solve_methods = [
@@ -17,6 +18,12 @@ def test_solve(self):
         with self.assertRaises(TypeError):
             utils.solve(c, None)
 
+        with self.assertRaises(ValueError):
+            utils.solve(None, "INVALID SOLVER")
+
+        with self.assertRaises(ValueError):
+            utils.solve(None, MockSolver)
+
         with self.assertRaises(ValueError):
             utils.solve(None, self.solve_methods[0])
 
@@ -47,7 +54,7 @@ def test_solve(self):
     def test_pprint(self):
         c = Cube()
         with self.assertRaises(ValueError):
-            utils.solve(None)
+            utils.pprint(None)
 
         with self.assertRaises(ValueError):
-            utils.solve(1)
+            utils.pprint(1)