✔️ refactor tests to use pytest

Author: Charizhardt

src/test_bpmn.py

@@ -1,4 +1,5 @@
 from processpiper import ProcessMap, EventType, ActivityType, GatewayType
+from util_test import get_test_file_path
 
 
 def test_case01():
@@ -16,11 +17,8 @@ def test_case01():
         process_map.set_footer("My footer", font_size=20)
 
         process_map.draw()
-        output_file = "images/test/test_appearance01.png"
+
+        output_file = get_test_file_path("test_appearance_01.png")
         process_map.save(output_file)
         output_file = output_file.replace(".png", ".bpmn")
         process_map.export_to_bpmn(output_file)
-
-
-if __name__ == "__main__":
-    test_case01()

src/test_promo.py

@@ -1,9 +1,10 @@
 from processpiper import ProcessMap, EventType, ActivityType, GatewayType
+from util_test import get_test_file_path
 
 
 def test_case01():
     with ProcessMap(
-        "Pizza Order Process", colour_theme="BLUEMOUNTAIN"
+            "Pizza Order Process", colour_theme="BLUEMOUNTAIN"
     ) as my_process_map:
         with my_process_map.add_lane("Customer") as lane1:
             start = lane1.add_element("start", EventType.START)
@@ -35,13 +36,16 @@ def test_case01():
             receive_pizza.connect(end)
         my_process_map.set_footer(" Generated by Process Piper")
         my_process_map.draw()
-        my_process_map.save("images/test/pizza-order-process.png")
-        my_process_map.export_to_bpmn("images/test/pizza-order-process.bpmn")
+
+        output_png = get_test_file_path("test_case_01.png")
+        output_bpmn = get_test_file_path("test_case_01.bpmn")
+        my_process_map.save(output_png)
+        my_process_map.export_to_bpmn(output_bpmn)
 
 
 def test_case02():
     with ProcessMap(
-        "Sample Test Process", colour_theme="BLUEMOUNTAIN"
+            "Sample Test Process", colour_theme="BLUEMOUNTAIN"
     ) as my_process_map:
         with my_process_map.add_lane("End User") as lane1:
             start = lane1.add_element("Start", EventType.START)
@@ -68,10 +72,8 @@ def test_case02():
 
         my_process_map.set_footer("Generated by ProcessPiper")
         my_process_map.draw()
-        my_process_map.save("images/test/my_process_map.png")
-        my_process_map.export_to_bpmn("images/test/my_process_map.bpmn")
-
 
-if __name__ == "__main__":
-    test_case01()
-    test_case02()
+        output_png = get_test_file_path("test_case_02.png")
+        output_bpmn = get_test_file_path("test_case_02.bpmn")
+        my_process_map.save(output_png)
+        my_process_map.export_to_bpmn(output_bpmn)