fix constants for graph performance tests

Thijss · Thijss · commit 62c6cde1c7e2 · 2025-02-03T21:26:29.000+01:00
Signed-off-by: Thijs Baaijen &lt;13253091+Thijss@users.noreply.github.com&gt;
diff --git a/tests/performance/_constants.py b/tests/performance/_constants.py
@@ -14,9 +14,9 @@
 }
 
 GRAPH_SETUP_CODES = {
-    "rustworkx": "from power_grid_model_ds.model.grids.base import Grid;"
-    + "from power_grid_model_ds.data_source.generator.grid_generators import RadialGridGenerator;"
-    + "from power_grid_model_ds.model.graphs.models import RustworkxGraphModel;"
+    "rustworkx": "from power_grid_model_ds import Grid;"
+    + "from power_grid_model_ds.generators import RadialGridGenerator;"
+    + "from power_grid_model_ds.graph_models import RustworkxGraphModel;"
     + "grid=RadialGridGenerator(nr_nodes={graph_size}, grid_class=Grid, graph_model=RustworkxGraphModel).run()",
 }
 
diff --git a/tests/performance/graph_performance_tests.py b/tests/performance/graph_performance_tests.py
@@ -7,7 +7,6 @@
 # pylint: disable=missing-function-docstring
 
 GRAPH_SIZES = [100, 500, 1000, 5000]
-GRAPH_SIZES = [100, 500, 1000]
 
 
 def perftest_initialize():

Original file line number	Diff line number	Diff line change
`@@ -14,9 +14,9 @@`
`14`	`14`	`}`
`15`	`15`
`16`	`16`	`GRAPH_SETUP_CODES = {`
`17`		`- "rustworkx": "from power_grid_model_ds.model.grids.base import Grid;"`
`18`		`- + "from power_grid_model_ds.data_source.generator.grid_generators import RadialGridGenerator;"`
`19`		`- + "from power_grid_model_ds.model.graphs.models import RustworkxGraphModel;"`
	`17`	`+ "rustworkx": "from power_grid_model_ds import Grid;"`
	`18`	`+ + "from power_grid_model_ds.generators import RadialGridGenerator;"`
	`19`	`+ + "from power_grid_model_ds.graph_models import RustworkxGraphModel;"`
`20`	`20`	`+ "grid=RadialGridGenerator(nr_nodes={graph_size}, grid_class=Grid, graph_model=RustworkxGraphModel).run()",`
`21`	`21`	`}`
`22`	`22`