-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_solver.py
76 lines (67 loc) · 2.57 KB
/
test_solver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/python
import unittest
import solver_2013CSB1029 as solver
import rubik
import sys
class TestSolver(unittest.TestCase):
def testShortestPath0(self):
"""Length 0 path."""
start = rubik.I
end = rubik.I
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 0)
def testShortestPath1(self):
"""Length 1 path."""
start = rubik.I
end = rubik.perm_apply(rubik.F, start)
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 1)
self.assertEqual(ans, [rubik.F])
def testShortestPath2(self):
"""Length 2 path."""
start = rubik.I
middle = rubik.perm_apply(rubik.F, start)
end = rubik.perm_apply(rubik.L, middle)
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 2)
self.assertEqual(ans, [rubik.F, rubik.L])
def testShortestPath3(self):
"""Length 3 path."""
start = rubik.I
middle1 = rubik.perm_apply(rubik.F, start)
middle2 = rubik.perm_apply(rubik.F, middle1)
end = rubik.perm_apply(rubik.Li, middle2)
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 3)
self.assertGoodPath(start, end, ans)
def testShortestPath4(self):
"""Length 4 path."""
start = rubik.I
middle1 = rubik.perm_apply(rubik.F, start)
middle2 = rubik.perm_apply(rubik.L, middle1)
middle3 = rubik.perm_apply(rubik.F, middle2)
end = rubik.perm_apply(rubik.L, middle3)
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 4)
self.assertGoodPath(start, end, ans)
def testshortestPath14(self):
"""Length 14 path."""
start = (6, 7, 8, 20, 18, 19, 3, 4, 5, 16, 17, 15, 0, 1, 2, 14, 12, 13, 10, 11, 9, 21, 22, 23)
end = rubik.I
ans = solver.shortest_path(start, end)
self.assertEqual(len(ans), 14)
self.assertGoodPath(start, end, ans)
def testshortestPathBad(self):
"""No solution."""
start = (7, 8, 6, 20, 18, 19, 3, 4, 5, 16, 17, 15, 0, 1, 2, 14, 12, 13, 10, 11, 9, 21, 22, 23)
end = rubik.I
ans = solver.shortest_path(start, end)
self.assertEqual(ans, None)
def assertGoodPath(self, start, end, path):
current = start
for move in path:
current = rubik.perm_apply(move, current)
self.assertEqual(current, end)
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(TestSolver)
unittest.TextTestRunner(verbosity=2).run(suite)