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))