Add support for custom stage typesHEADmain

1 files changed, 7 insertions, 7 deletions

diff --git a/tests/integration_test.rs b/tests/integration_test.rs
index 122ed5a..7ae228f 100644
--- a/tests/integration_test.rs
+++ b/tests/integration_test.rs
@@ -6,7 +6,7 @@ fn test_render_simple_map_to_png() {
     let map = wmap_parser::parse(source);
     let config = Configuration::default();
 
-    let result = render_to_png(&map, StageType::default(), &config);
+    let result = render_to_png(&map, &StageType::default(), &config);
     assert!(result.is_ok());
 
     let png_data = result.unwrap();
@@ -22,7 +22,7 @@ fn test_render_simple_map_to_svg() {
     let map = wmap_parser::parse(source);
     let config = Configuration::default();
 
-    let result = render_to_svg(&map, StageType::default(), &config);
+    let result = render_to_svg(&map, &StageType::default(), &config);
     assert!(result.is_ok());
 
     let svg_data = result.unwrap();
@@ -36,7 +36,7 @@ fn test_render_with_all_shapes() {
     let map = wmap_parser::parse(source);
     let config = Configuration::default();
 
-    let result = render_to_png(&map, StageType::default(), &config);
+    let result = render_to_png(&map, &StageType::default(), &config);
     assert!(result.is_ok());
 }
 
@@ -47,9 +47,9 @@ fn test_render_with_different_stage_types() {
     let config = Configuration::default();
 
     // Test a few different stage types
-    assert!(render_to_png(&map, StageType::Activities, &config).is_ok());
-    assert!(render_to_png(&map, StageType::Cynefin, &config).is_ok());
-    assert!(render_to_png(&map, StageType::Practice, &config).is_ok());
+    assert!(render_to_png(&map, &StageType::Activities, &config).is_ok());
+    assert!(render_to_png(&map, &StageType::Cynefin, &config).is_ok());
+    assert!(render_to_png(&map, &StageType::Practice, &config).is_ok());
 }
 
 #[test]
@@ -58,6 +58,6 @@ fn test_render_empty_map() {
     let map = wmap_parser::parse(source);
     let config = Configuration::default();
 
-    let result = render_to_png(&map, StageType::default(), &config);
+    let result = render_to_png(&map, &StageType::default(), &config);
     assert!(result.is_ok());
 }