diff --git a/src/ex2.py b/src/ex2.py index 2ce418ef..68d0831a 100644 --- a/src/ex2.py +++ b/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() diff --git a/tests/test_ex2.py b/tests/test_ex2.py new file mode 100644 index 00000000..03eff0dd --- /dev/null +++ b/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