20 lines
525 B
Python
20 lines
525 B
Python
import pytest
|
|
from solution import Solution
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"weights,days,expected",
|
|
[
|
|
([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 5, 15),
|
|
([3, 2, 2, 4, 1, 4], 3, 6),
|
|
([1, 2, 3, 1, 1], 4, 3),
|
|
([1], 1, 1),
|
|
([5, 5, 5, 5], 4, 5),
|
|
([5, 5, 5, 5], 1, 20),
|
|
([10, 50, 100, 100, 50, 10], 1, 320),
|
|
([10, 50, 100, 100, 50, 10], 6, 100),
|
|
],
|
|
)
|
|
def test_ship_within_days(weights, days, expected):
|
|
assert Solution().shipWithinDays(weights, days) == expected
|