def test_robotVacuum_2(self):
room = [[1, 8, 5, 4],
[3, 4, 1, 0],
[3, 5, 2, 1],
[0, 2, 1, 4]]
maxDirt = 3
assert robotVacuum(room, maxDirt) == [8, 8, 13, 15]
def test_robotVacuum_3(self):
room = [[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0]]
maxDirt = 3
assert robotVacuum(room, maxDirt) == [0, 0, 0, 0]
def test_robotVacuum_5(self):
room = [[0 for column in range(500)] for row in range(500)]
maxDirt = 1
assert robotVacuum(room, maxDirt) == [0, 0, 0, 0]
def test_robotVacuum_4(self):
room = [[5000 for column in range(3)] for row in range(3)]
maxDirt = 1
assert robotVacuum(room, maxDirt) == [20000, 10000, 5000, 10000]
def test_robotVacuum_1(self):
board = [[0, 5, 4, 10, 0, 8],
[7, 2, 7, 4, 7, 2],
[2, 2, 10, 1, 7, 6]]
maxDirt = 10
assert robotVacuum(board, maxDirt) == [18, 27, 13, 26]