15 lines
310 B
Python
15 lines
310 B
Python
import pytest
|
|
from solution import Solution
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"nums,target,expected",
|
|
[
|
|
([2, 7, 11, 15], 9, [0, 1]),
|
|
([3, 2, 4], 6, [1, 2]),
|
|
([3, 3], 6, [0, 1]),
|
|
],
|
|
)
|
|
def test_two_sum(nums, target, expected):
|
|
assert Solution().twoSum(nums, target) == expected
|