feat: add ex2.py

src/ex2.py

@@ -49,3 +49,25 @@
 #
 #   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]
+    else:
+        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
+from src import ex2
+
+def test_week_number_1(mocker: MockerFixture) -> None:
+
+    #arrange 
+    mock_number_return = '1'
+    mocker.patch.object(ex2, "input", return_value=mock_number_return)
+
+    #act
+    res = ex2.week_number()
+
+    #assert
+    assert res == 'Monday'
+    assert type(res) is str
+
+def test_week_number_2(mocker: MockerFixture) -> None:
+
+    #arrange 
+    mock_number_return = '9'
+    mocker.patch.object(ex2, "input", return_value=mock_number_return)
+
+    #act
+    res = ex2.week_number()
+
+    #assert
+    assert res == ''
+    assert type(res) is str