From 59930b2df6e498fad15b026a435e0847bb34f6ea Mon Sep 17 00:00:00 2001 From: Marco Gorelli <33491632+MarcoGorelli@users.noreply.github.com> Date: Thu, 11 Apr 2024 08:33:17 +0100 Subject: [PATCH] try getting 100% coverage --- .../unit/functions/business/test_business_day_count.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/py-polars/tests/unit/functions/business/test_business_day_count.py b/py-polars/tests/unit/functions/business/test_business_day_count.py index 94ce1226c15b..f1ce93268810 100644 --- a/py-polars/tests/unit/functions/business/test_business_day_count.py +++ b/py-polars/tests/unit/functions/business/test_business_day_count.py @@ -109,14 +109,14 @@ def test_business_day_count_schema() -> None: def test_business_day_count_w_holidays() -> None: df = pl.DataFrame( { - "start": [date(2020, 1, 1), date(2020, 1, 2)], - "end": [date(2020, 1, 2), date(2020, 1, 10)], + "start": [date(2020, 1, 1), date(2020, 1, 2), date(2020, 1, 2)], + "end": [date(2020, 1, 2), date(2020, 1, 10), date(2020, 1, 9)], } ) result = df.select( business_day_count=pl.business_day_count( - "start", "end", holidays=[date(2020, 1, 1)] + "start", "end", holidays=[date(2020, 1, 1), date(2020, 1, 9)] ), )["business_day_count"] - expected = pl.Series("business_day_count", [0, 6], pl.Int32) + expected = pl.Series("business_day_count", [0, 5, 5], pl.Int32) assert_series_equal(result, expected)