loopover/test_loopover.py at master · AndreiToroplean/loopover · GitHub

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
class TestLoopoverPuzzle:
    def test_get_solution(self):
        from loopover_puzzle import LoopoverPuzzle
        loopover_puzzle = LoopoverPuzzle.from_shape((10, 10), randomize=True)
        loopover_puzzle_solved = LoopoverPuzzle.from_shape((10, 10), randomize=True)
        loopover_puzzle.define_solved_perm(loopover_puzzle_solved)
        solution = loopover_puzzle.get_solution()
        loopover_puzzle.apply_action(solution)
        assert loopover_puzzle.is_solved

    def test_apply_tri_rot(self):
        from rotcomp import Rot
        from loopover_puzzle import LoopoverPuzzle
        loopover_puzzle_a = LoopoverPuzzle.from_shape((10, 10), randomize=True)
        loopover_puzzle_b = loopover_puzzle_a.copy()
        tri_rot = Rot.from_random(max_index=loopover_puzzle_a.n_pieces, len_=3)
        loopover_puzzle_a._rot_directly(tri_rot)
        loopover_puzzle_b.rot(tri_rot)
        assert loopover_puzzle_a.has_equal_board(loopover_puzzle_b)


class TestRotComp:
    def test_compressed(self):
        from rotcomp import RotComp
        from linear_puzzle import LinearPuzzle
        rotcomp = RotComp.from_random(4, max_index=16)
        linear_puzzle_a = LinearPuzzle.from_rotcomp(rotcomp)
        linear_puzzle_b = linear_puzzle_a.copy()
        linear_puzzle_a.apply_action(rotcomp)
        linear_puzzle_b.apply_action(rotcomp.compressed())
        assert linear_puzzle_a.has_equal_board(linear_puzzle_b)


class TestMoveComp:
    def test_compressed(self):
        from movecomp import MoveComp
        from loopover_puzzle import LoopoverPuzzle
        loopover_puzzle_a = LoopoverPuzzle.from_shape((10, 10), randomize=True)
        loopover_puzzle_b = loopover_puzzle_a.copy()
        movecomp = loopover_puzzle_a.get_random_movecomp(100)
        loopover_puzzle_a.move(movecomp.compressed())
        loopover_puzzle_b.move(movecomp)
        assert loopover_puzzle_a.has_equal_board(loopover_puzzle_b)

    def test_as_strs(self):
        from movecomp import MoveComp
        from loopover_puzzle import LoopoverPuzzle
        loopover_puzzle = LoopoverPuzzle.from_shape((10, 10), randomize=True)
        movecomp = loopover_puzzle.get_random_movecomp(10)
        assert MoveComp.from_strs(movecomp.as_strs) == movecomp


class TestLoopover:
    def test_loopover(self):
        from loopover import loopover
        from loopover_puzzle import LoopoverPuzzle
        board = [list(x) for x in ["abc", "dfe", "gih"]]
        solved_board = [list(x) for x in ["abc", "def", "ghi"]]
        loopover_puzzle = LoopoverPuzzle(board)
        loopover_puzzle_solved = LoopoverPuzzle(solved_board)
        loopover_puzzle.define_solved_perm(loopover_puzzle_solved)
        solution = loopover(board, solved_board)
        print(solution)
        loopover_puzzle.apply_action_strs(solution)
        assert loopover_puzzle.is_solved