feat: add ex2.py

src/ex2.py

@@ -49,3 +49,31 @@
 #
 #   return 0;
 # }
+
+
+def week_number():
+    """
+    Returns the day of the week
+    """
+    number = int(input("Enter week number(1-7): "))
+    week_day = [
+        "Monday",
+        "Tuesday",
+        "Wednesday",
+        "Thursday",
+        "Friday",
+        "Saturday",
+        "Sunday",
+    ]
+
+    if number > 0 and number < 8:
+        return week_day[number - 1]
+    return ''
+
+
+if __name__ == "__main__":
+    res = week_number()
+    if res != "":
+        print(res)
+    else:
+        print("Invalid input")

tests/test_ex2.py

@@ -0,0 +1,28 @@
+from pytest_mock import MockerFixture
+import pytest
+from typing import List
+from src import ex2
+
+tests: List[dict] = [
+    { "mock_value": '1', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Monday' },
+    { "mock_value": '2', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Tuesday' },
+    { "mock_value": '3', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Wednesday' },
+    { "mock_value": '4', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Thursday' },
+    { "mock_value": '5', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Friday' },
+    { "mock_value": '6', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Saturday' },
+    { "mock_value": '7', "mock_function": "input", "res": ex2.week_number, "expected_res": 'Sunday' },
+    { "mock_value": '9', "mock_function": "input", "res": ex2.week_number, "expected_res": '' }
+]
+
+@pytest.mark.parametrize("test", tests)
+def test_week_number(mocker: MockerFixture, test: dict) -> None:
+
+    #arrange 
+    mocker.patch.object(ex2, test['mock_function'], return_value=test["mock_value"])
+
+    #act
+    res = test["res"]()
+
+    #assert
+    assert res == test["expected_res"]
+    assert type(res) is str