def test_4x4(self): array = [[1, 1, 2, 4], [2, 2, 1, 2], [2, 1, 2, 1], [2, 1, 2, 1]] max_gold_amount, route_string = castle_walk(array) self.assertEqual(max_gold_amount, 12) self.assertEqual(route_string, 'R R R D D D')
def test_4x5(self): array = [[1, 1, 2, 4, 1], [2, 2, 1, 2, 1], [2, 1, 2, 1, 2], [2, 1, 2, 1, 1]] max_gold_amount, route_string = castle_walk(array) self.assertEqual(max_gold_amount, 14) self.assertIn(route_string, ('R R R D D R D', 'R R R D R D D'))
def test_3x4(self): array = [[1, 1, 2, 1], [2, 2, 1, 1], [2, 1, 2, 1]] max_gold_amount, route_string = castle_walk(array) self.assertEqual(max_gold_amount, 9) self.assertIn(route_string, ('D R R D R', 'D R D R R', 'D D R R R'))