Amazing (Python): Code Cleanup

* Make more use of a Maze class
* Create a bigger maze in the test
* Group row/col variables in Position class
* Group direction variables in an Enum

02_Amazing/python/test_amazing.py

@@ -1,8 +1,13 @@
+import io
 import pytest
-from amazing import build_maze, welcome_header
+from _pytest.capture import CaptureFixture
+from _pytest.monkeypatch import MonkeyPatch
+
+from amazing import build_maze, welcome_header, main
 
 
-def test_welcome_header(capsys) -> None:
+def test_welcome_header(capsys: CaptureFixture[str]) -> None:
+    capsys.readouterr()
     welcome_header()
     out, err = capsys.readouterr()
     assert out == (
@@ -25,3 +30,29 @@ def test_welcome_header(capsys) -> None:
 def test_build_maze(width: int, length: int) -> None:
     with pytest.raises(AssertionError):
         build_maze(width, length)
+
+
+@pytest.mark.parametrize(
+    ("width", "length"),
+    [
+        (3, 3),
+        (10, 10),
+    ],
+)
+def test_main(monkeypatch: MonkeyPatch, width: int, length: int) -> None:
+    monkeypatch.setattr(
+        "sys.stdin",
+        io.StringIO(f"{width},{length}"),
+    )
+    main()
+
+
+def test_main_error(monkeypatch: MonkeyPatch) -> None:
+    width = 1
+    length = 2
+
+    monkeypatch.setattr(
+        "sys.stdin",
+        io.StringIO(f"{width},{length}\n3,3"),
+    )
+    main()