mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
39 lines
1012 B
Python
39 lines
1012 B
Python
from nose.tools import assert_equal
|
|
|
|
|
|
class TestGridPath(object):
|
|
|
|
def test_grid_path(self):
|
|
grid = Grid()
|
|
assert_equal(grid.find_path(None), None)
|
|
assert_equal(grid.find_path([[]]), None)
|
|
max_rows = 8
|
|
max_cols = 4
|
|
matrix = [[1] * max_cols for _ in range(max_rows)]
|
|
matrix[1][1] = 0
|
|
matrix[2][2] = 0
|
|
matrix[3][0] = 0
|
|
matrix[4][2] = 0
|
|
matrix[5][3] = 0
|
|
matrix[6][1] = 0
|
|
matrix[6][3] = 0
|
|
matrix[7][1] = 0
|
|
result = grid.find_path(matrix)
|
|
expected = [(0, 0), (1, 0), (2, 0),
|
|
(2, 1), (3, 1), (4, 1),
|
|
(5, 1), (5, 2), (6, 2),
|
|
(7, 2), (7, 3)]
|
|
assert_equal(result, expected)
|
|
matrix[7][2] = 0
|
|
result = grid.find_path(matrix)
|
|
assert_equal(result, None)
|
|
print('Success: test_grid_path')
|
|
|
|
|
|
def main():
|
|
test = TestGridPath()
|
|
test.test_grid_path()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |