def test_main(self):
stdout, stderr = sys.stdout, sys.stderr
sys.stdout, sys.stderr = StringIO(), StringIO()
with self.assertRaises(SystemExit):
utils.main([])
with self.assertRaises(SystemExit):
utils.main(['-c'])
with self.assertRaises(SystemExit):
utils.main(['-h'])
with self.assertRaises(SystemExit):
utils.main(['-i'])
# Discard stdout
sys.stdout = StringIO()
for method in self.solve_methods:
for i in range(10):
c = Cube()
ref_solution = method(c).solution()
utils.main(['--cube', c.to_naive_cube().get_cube()])
utils.main(['-i', c.to_naive_cube().get_cube()])
# Restore stdout and stderr
sys.stdout, sys.stderr = stdout, stderr
def test_solve(self):
c = Cube()
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])
with self.assertRaises(ValueError):
utils.solve(1, self.solve_methods[0])
for method in self.solve_methods:
for i in range(10):
c = Cube()
ref_solution = method(c).solution()
s1 = utils.solve(c, method)
self.assertEqual(ref_solution,
s1,
msg="Failed with Cubie.Cube and method %s" %
method.__class__.__name__)
# Test with NaiveCube
s2 = utils.solve(c.to_naive_cube(), method)
self.assertEqual(ref_solution,
s2,
msg="Failed with Cubie.Cube and method %s" %
method.__class__.__name__)
# Test with string representation
s3 = utils.solve(c.to_naive_cube().get_cube(), method)
self.assertEqual(ref_solution,
s3,
msg="Failed with Cubie.Cube and method %s" %
method.__class__.__name__)