leetcode/array/remove_element.py

from typing import List
import pytest

def removeElement(nums: List[int], val: int):
    i = 0
    while i < len(nums):
        if nums[i] == val:
            nums.remove(nums[i])
        else:
            i += 1
    return len(nums)


@pytest.mark.parametrize(
    "nums, val, expect",
    [
        ([-2, 3, 1, 5], 3, 3),
        ([2, 3, 3, 2], 3, 2)
    ]
)
def test_cases(nums, val, expect):
    assert removeElement(nums, val) == expect