Update visualizer builder

2 files changed, 14 insertions, 8 deletions

diff --git a/subprojects/store-dse/src/test/java/tools/refinery/store/dse/CRAExamplesTest.java b/subprojects/store-dse/src/test/java/tools/refinery/store/dse/CRAExamplesTest.java
index f1e90280..23f0e703 100644
--- a/subprojects/store-dse/src/test/java/tools/refinery/store/dse/CRAExamplesTest.java
+++ b/subprojects/store-dse/src/test/java/tools/refinery/store/dse/CRAExamplesTest.java
@@ -188,7 +188,13 @@ class CRAExamplesTest {
                                                 .queries(feature, assignFeaturePreconditionHelper, assignFeaturePrecondition,
                                                                 deleteEmptyClassPrecondition, createClassPreconditionHelper, createClassPrecondition,
                                                                 moveFeaturePrecondition))
-                                .with(ModelVisualizerAdapter.builder())
+                                .with(ModelVisualizerAdapter.builder()
+                                                .withOutputpath("test_output")
+                                                .withFormat(FileFormat.DOT)
+                                                .withFormat(FileFormat.SVG)
+                                                .saveStates()
+                                                .saveDesignSpace()
+                                )
                                 .with(DesignSpaceExplorationAdapter.builder()
                                                 .transformations(assignFeatureRule, deleteEmptyClassRule, createClassRule, moveFeatureRule)
 //                                              .strategy(new DepthFirstStrategy(3).continueIfHardObjectivesFulfilled()
@@ -271,9 +277,6 @@ class CRAExamplesTest {
 
                 var states = dseAdapter.explore();
                 System.out.println("states size: " + states.size());
-                System.out.println("states: " + states);
-                var visualizer = model.getAdapter(ModelVisualizerAdapter.class);
-                visualizer.renderDesignSpace("test_output", FileFormat.SVG);
         }
 
 }
diff --git a/subprojects/store-dse/src/test/java/tools/refinery/store/dse/DebugTest.java b/subprojects/store-dse/src/test/java/tools/refinery/store/dse/DebugTest.java
index 553510c8..88c4f152 100644
--- a/subprojects/store-dse/src/test/java/tools/refinery/store/dse/DebugTest.java
+++ b/subprojects/store-dse/src/test/java/tools/refinery/store/dse/DebugTest.java
@@ -90,7 +90,13 @@ class DebugTest {
                                 .symbols(classModel, classElement, feature, isEncapsulatedBy, encapsulates, classes, features)
                                 .with(ViatraModelQueryAdapter.builder()
                                                 .queries(createClassPrecondition, createFeaturePrecondition))
-                                .with(ModelVisualizerAdapter.builder())
+                                .with(ModelVisualizerAdapter.builder()
+                                                .withOutputpath("test_output")
+                                                .withFormat(FileFormat.DOT)
+                                                .withFormat(FileFormat.SVG)
+                                                .saveStates()
+                                                .saveDesignSpace()
+                                )
                                 .with(DesignSpaceExplorationAdapter.builder()
                                                 .transformations(createClassRule, createFeatureRule)
                                                 .strategy(new DepthFirstStrategy(4).continueIfHardObjectivesFulfilled()
@@ -113,10 +119,7 @@ class DebugTest {
 
 
                 var states = dseAdapter.explore();
-                var visualizer = model.getAdapter(ModelVisualizerAdapter.class);
-                visualizer.renderDesignSpace("test_output", FileFormat.SVG);
                 System.out.println("states size: " + states.size());
-                System.out.println("states: " + states);
 
         }
 }