๐Ÿ“ฆ schubart / LeetCode

๐Ÿ“„ lc_0027_test.py ยท 18 lines
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18from typing import List

from lc_0027 import Solution


def test_examples():
    def validate(nums: List[int], val):
        expected = [n for n in nums if n != val]

        length = Solution().removeElement(nums, val)

        assert sorted(nums[:length]) == sorted(expected)

    validate([3, 2, 2, 3], 3)
    validate([0, 1, 2, 2, 3, 0, 4, 2], 2)
    validate([0, 1, 2, 2, 3, 0, 4, 2], 10)
    validate([], 0)