bites 197 - datetime

Author: [email protected]

.gitignore

@@ -69,3 +69,4 @@
 /116/README.md
 /187/README.md
 /148/README.md
+/197/README.md

197/mday.py

@@ -0,0 +1,15 @@
+from datetime import date, timedelta
+
+
+def get_mothers_day_date(year):
+    """Given the passed in year int, return the date Mother's Day
+       is celebrated assuming it's the 2nd Sunday of May."""
+    day = date(year=year, month=5, day=1)
+    while 1:
+        if day.weekday() == 6:
+            day += timedelta(days=7)
+            break
+        day += timedelta(days=1)
+    return day
+
+# print(get_mothers_day_date(2014))

197/test_mday.py

@@ -0,0 +1,26 @@
+from datetime import date
+
+import pytest
+
+from mday import get_mothers_day_date
+
+
+def test_check_type():
+    assert type(get_mothers_day_date(2019)) == date
+
+
+@pytest.mark.parametrize('arg,expected', [
+    (2014, date(2014, 5, 11)),
+    (2015, date(2015, 5, 10)),
+    (2016, date(2016, 5, 8)),
+    (2017, date(2017, 5, 14)),
+    (2018, date(2018, 5, 13)),
+    (2019, date(2019, 5, 12)),
+    (2020, date(2020, 5, 10)),
+    (2021, date(2021, 5, 9)),
+    (2022, date(2022, 5, 8)),
+    (2023, date(2023, 5, 14)),
+    (2024, date(2024, 5, 12)),
+])
+def test_return_various_years(arg, expected):
+    assert get_mothers_day_date(arg) == expected