def test_taking_group_stones(self):
board = np.array([[0, 1, 0], [1, -1, 1], [1, -1, 1], [0, 1, 0]])
target_group = [(1, 1), (1, 2)]
group = capture_group(1, 1, board)
self.assertEqual(group, target_group)
target_group = [(1, 2), (1, 1)]
group = capture_group(1, 2, board)
self.assertEqual(group, target_group)
def test_taking_group_sucide(self):
board = np.array([[-1, 1, 0], [1, 0, 0], [0, 0, 0]])
target_group = [(0, 0)]
group = capture_group(0, 0, board)
self.assertEqual(group, target_group)
board = np.array([[-1, -1, 1], [1, 1, 0], [0, 0, 0]])
target_group = [(0, 0), (1, 0)]
group = capture_group(0, 0, board)
self.assertEqual(group, target_group)
target_group = [(1, 0), (0, 0)]
group = capture_group(1, 0, board)
self.assertEqual(group, target_group)
def test_circle_group(self):
board = np.array([[0, 1, 1, 1, 0], [1, -1, -1, -1,
1], [1, -1, 1, -1, 1],
[1, -1, -1, -1, 1], [0, 1, 1, 1, 0]])
target_group = [(1, 1), (2, 1), (3, 1), (3, 2), (3, 3), (2, 3), (1, 3),
(1, 2)]
self.assertEqual(len(target_group), 8)
for x, y in target_group:
group = capture_group(x, y, board)
self.assertEqual(sorted(group), sorted(target_group))