def test_generate_board_triangle_lower(self):
board = hg.generate_triangular_board(5, mirrored=True)
expected_board = [
[0, 0, 0, 0, 1],
[0, 0, 0, 1, 1],
[0, 0, 1, 1, 1],
[0, 1, 1, 1, 1],
[1, 1, 1, 1, 1],
]
self.assertEqual(board, expected_board)
def test_generate_board_triangle(self):
board = hg.generate_triangular_board(5)
expected_board = [
[1, 1, 1, 1, 1],
[1, 1, 1, 1, 0],
[1, 1, 1, 0, 0],
[1, 1, 0, 0, 0],
[1, 0, 0, 0, 0],
]
self.assertEqual(board, expected_board)
