From f7e008539261fff2a7f54c0f8a6bce88d9d2cdad Mon Sep 17 00:00:00 2001 From: Thijs Baaijen <13253091+Thijss@users.noreply.github.com> Date: Mon, 3 Feb 2025 21:15:36 +0100 Subject: [PATCH] fix graph imports for performance tests Signed-off-by: Thijs Baaijen <13253091+Thijss@users.noreply.github.com> --- tests/performance/_constants.py | 6 +++--- tests/performance/graph_performance_tests.py | 10 +++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/performance/_constants.py b/tests/performance/_constants.py index f114754..98c2754 100644 --- 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 index 8b242df..50b1aae 100644 --- a/tests/performance/graph_performance_tests.py +++ b/tests/performance/graph_performance_tests.py @@ -48,9 +48,9 @@ def perftest_add_node(): if __name__ == "__main__": - perftest_initialize() - perftest_set_feeder_ids() + # perftest_initialize() + # perftest_set_feeder_ids() perftest_get_components() - perftest_delete_node() - perftest_add_node() - perftest_from_arrays() + # perftest_delete_node() + # perftest_add_node() + # perftest_from_arrays()