14 lines
275 B
Python
14 lines
275 B
Python
import pytest
|
|
from solution import Solution
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"nums, k, expected",
|
|
[
|
|
([7, 2, 5, 10, 8], 2, 18),
|
|
([1, 2, 3, 4, 5], 2, 9),
|
|
],
|
|
)
|
|
def test_split_array(nums, k, expected):
|
|
assert Solution().splitArray(nums, k) == expected
|