theovoss / Chess

        assert new_end_locations == expected_positions

    def test_cant_jump_own_pieces_diagonally(self):

        start = (0, 2)

        middle = (1, 1)

        self.chess_board[start] = Mock(color=1)

        self.chess_board[middle] = Mock(color=1)

        self.chess_board[(2, 0)] = None

        starting_end_locations = [(1, 1), (2, 0)]

        expected_positions = [(1, 1)]

        new_end_locations = cant_jump_pieces(self.chess_board, start, None, starting_end_locations, Mock())

        assert new_end_locations == expected_positions

    def test_cant_jump_pieces_divide_by_one_error(self):

        start = (2, 2)

        ends = ends_on_enemy(self.chess_board, start, None, potential_end_locations, Mock())

        assert ends == []

    def test_ends_on_enemy_players_piece(self):

        start = (0, 0)

        self.chess_board[start] = Mock(color=1)

        potential_end_locations = [(1, 0), (2, 0), (2, 2)]

        end = None

        for end in potential_end_locations:

            self.chess_board[end] = Mock(color=1)

        self.chess_board[end] = Mock(color=2)

        ends = ends_on_enemy(self.chess_board, start, None, potential_end_locations, Mock())

        assert ends == [(2, 2)]

    def test_doesnt_land_on_piece(self):

        start = (1, 1)