feat: implement week day selection and add tests

src/ex1.py

@@ -47,3 +47,31 @@
 #
 #   return 0;
 # }
+
+
+def main():
+    try:
+        week = int(input("Enter week number(1-7):"))
+    except ValueError:
+        print("Invalid input! Please enter a valid integer between 1-7.")
+        return
+    if week == 1:
+        print("Monday")
+    elif week == 2:
+        print("Tuesday")
+    elif week == 3:
+        print("Wednesday")
+    elif week == 4:
+        print("Thursday")
+    elif week == 5:
+        print("Friday")
+    elif week == 6:
+        print("Saturday")
+    elif week == 7:
+        print("Sunday")
+    else:
+        print("Invalid input! Please enter week number between 1-7.")
+
+
+if __name__ == "__main__":  # pragma: no cover
+    main()

tests/test_ex1.py

@@ -0,0 +1,26 @@
+from src.ex1 import main
+import builtins
+
+data = [
+    (1, "Monday\n"),
+    (2, "Tuesday\n"),
+    (3, "Wednesday\n"),
+    (4, "Thursday\n"),
+    (5, "Friday\n"),
+    (6, "Saturday\n"),
+    (7, "Sunday\n"),
+    (8, "Invalid input! Please enter week number between 1-7.\n"),
+    (0, "Invalid input! Please enter week number between 1-7.\n"),
+    (-1, "Invalid input! Please enter week number between 1-7.\n"),
+    ("abc", "Invalid input! Please enter a valid integer between 1-7.\n"),
+]
+
+
+def test_main(monkeypatch, capsys):
+    for week_input, expected_output in data:
+        monkeypatch.setattr(builtins, "input", lambda _: str(week_input))
+
+        main()
+
+        captured = capsys.readouterr()
+        assert expected_output == captured.out