import unittest

from maze import parse_maze


class MazeTests(unittest.TestCase):
    def setUp(self):
        s = """\
1 0 0 0 1 0 0 0 
0 1 1 1 1 1 0 0 
0 1 0 1 1 0 0 0 
0 1 1 1 0 0 1 1 
1 1 0 0 0 1 0 0 
1 1 1 0 0 1 1 0 
1 0 1 0 0 1 0 1 
0 0 1 0 1 0 1 0 

1 0 0 0 0 1 0 0 
1 1 0 0 0 1 0 1 
1 0 1 0 1 1 1 1 
0 0 0 0 1 0 0 1 
0 0 0 0 1 0 1 1 
0 0 0 1 0 0 1 0 
1 0 1 0 1 0 0 0 
0 1 0 1 1 0 0 1 

1 1 1 0 0 0 0 1 
1 0 0 1 1 1 0 1 
1 0 0 1 1 0 1 0 
0 1 0 1 0 1 1 1 
1 0 1 1 1 1 0 0 
0 1 0 1 1 0 0 0 
1 1 0 1 0 0 0 1 
1 1 1 0 0 0 1 0 

1 1 0 0 1 0 1 1 
1 1 1 1 1 0 0 1 
1 1 0 0 1 0 1 1 
0 1 1 1 1 1 0 1 
0 1 1 0 0 0 1 0 
0 1 0 1 0 0 0 0 
1 1 0 1 0 1 1 0 
0 0 0 0 1 1 1 0 

1 0 0 0 1 1 1 0 
0 1 0 1 0 1 0 1 
1 0 0 1 0 0 0 0 
1 1 1 0 0 1 0 0 
0 1 1 0 0 0 1 0 
0 1 0 1 0 1 0 0 
1 0 0 1 1 1 0 1 
1 1 1 1 0 0 0 1 

1 0 1 1 1 0 0 0 
0 0 0 1 0 0 1 0 
0 1 1 0 0 0 0 1 
0 0 0 1 0 0 0 1 
1 0 1 1 0 1 1 0 
1 0 1 0 1 1 0 1 
0 0 0 0 0 0 1 1 
1 1 0 0 1 0 0 0 

0 0 0 0 0 0 1 1 
1 0 0 1 1 1 0 1 
0 1 0 1 1 1 1 1 
0 1 1 0 0 1 1 0 
1 1 1 0 1 1 1 1 
1 0 1 1 0 1 0 0 
0 0 0 1 0 0 1 0 
1 1 1 0 1 1 1 0 

1 1 1 1 0 0 1 0 
0 0 1 0 0 1 0 1 
0 1 1 1 0 1 0 0 
1 1 1 0 1 0 0 1 
0 0 1 0 1 0 0 0 
1 1 1 1 0 1 0 1 
1 0 1 1 1 1 1 0 
1 0 0 0 1 1 1 1 

"""
        self.m = parse_maze(s)

    def test_at_indices(self):
        assert self.m._data[7][0][0] == 0
        assert self.m._data[7][3][0] == 1
        assert self.m._data[7][0][1] == 0
        assert self.m._data[3][3][3] == 1
        assert self.m._data[5][2][6] == 1
        assert self.m._data[7][5][4] == 0