feat: refactor ex2.py to use an array for weekdays and implement main…

src/ex2.py

@@ -49,3 +49,26 @@
 #
 #   return 0;
 # }
+
+weekdays = [
+    "Monday",
+    "Tuesday",
+    "Wednesday",
+    "Thursday",
+    "Friday",
+    "Saturday",
+    "Sunday",
+]
+
+
+def main():
+    week = int(input("Enter week number(1-7): "))
+
+    if 1 <= week <= 7:
+        print(weekdays[week - 1])
+    else:
+        print("Invalid input! Please enter week number between 1-7.")
+
+
+if __name__ == "__main__":
+    main()

tests/test_ex2.py

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