updated output paths for tests to use moduel name

cracked-machine · Feb 4, 2024 · 8a86913 · 8a86913
1 parent 5d2ee10
commit 8a86913
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/tests/test_args.py b/tests/test_args.py
@@ -3,6 +3,8 @@
 import mmdiagram.generator
 import pathlib
 
+# Check the output report at /tmp/pytest/tests.test_args.md
+
 
 def test_no_args():
     with unittest.mock.patch('sys.argv',
@@ -42,7 +44,7 @@ def test_invalid_out_arg():
                               '0x10',
                               '0x10',
                               "-o",
-                              "/tmp/custom/myreport.txt"]):
+                              f"/tmp/pytest/{__name__}.txt"]):
         with pytest.raises(NameError):
             mmdiagram.generator.Diagram()
 
@@ -81,7 +83,7 @@ def test_valid_custom_out_arg():
                               '0x10',
                               '0x10',
                               "-o",
-                              "/tmp/custom/myreport.md"]):
+                              f"/tmp/pytest/{__name__}.md"]):
         mmdiagram.generator.Diagram()
-        assert pathlib.Path("/tmp/custom/myreport.md").exists()
-        assert pathlib.Path("/tmp/custom/myreport.png").exists()
+        assert pathlib.Path(f"/tmp/pytest/{__name__}.md").exists()
+        assert pathlib.Path(f"/tmp/pytest/{__name__}.png").exists()
diff --git a/tests/test_colours.py b/tests/test_colours.py
@@ -19,7 +19,7 @@ def test_no_more_colours_but_white():
                               '0x10',
                               '0x150',
                               "-o",
-                              "/tmp/custom/dupecolours.md"]):
+                              f"/tmp/pytest/{__name__}.md"]):
         # deliberately restrict the dict len=1 so we can confirm the error is raised.
         with unittest.mock.patch('mmdiagram.types.Region._remaining_colours',
                                  {"white": "#000000"}):
@@ -42,7 +42,7 @@ def test_no_more_colours_but_black():
                               '0x10',
                               '0x150',
                               "-o",
-                              "/tmp/custom/dupecolours.md"]):
+                              f"/tmp/pytest/{__name__}.md"]):
         # deliberately restrict the dict len=1 so we can confirm the error is raised.
         with unittest.mock.patch('mmdiagram.types.Region._remaining_colours',
                                  {"black": "#ffffff"}):
@@ -56,7 +56,7 @@ def test_only_one_lightslategr_y():
                              ['mmap_digram.diagram',
                               'kernel', '0x10', '0x60',
                               'rootfs', '0x50', '0x10'
-                              "-o", "/tmp/custom/onlytwocolours.md"],
+                              "-o", f"/tmp/pytest/{__name__}.md"],
                              'mmdiagram.types.Region._remaining_colours',
                              {"lightslategray": "#778899", "lightslategrey": "#778899"}):
         with pytest.raises(SystemExit):