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