From aef90a28ea347be9a9dd4818aed381d4d1ebe1c6 Mon Sep 17 00:00:00 2001 From: Josip Mrden Date: Tue, 20 Jun 2023 18:03:47 +0200 Subject: [PATCH] Add fixes to e2e --- tests/e2e/analyze_graph/optimize_indexes.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/e2e/analyze_graph/optimize_indexes.py b/tests/e2e/analyze_graph/optimize_indexes.py index 79ef65997..20cfd43a2 100644 --- a/tests/e2e/analyze_graph/optimize_indexes.py +++ b/tests/e2e/analyze_graph/optimize_indexes.py @@ -279,10 +279,8 @@ def test_same_avg_group_size_diff_distribution(connect): execute_and_fetch_all(cursor, "DROP INDEX ON :Label(id2);") -def test_given_supernode_when_expanding_then_expand_other_way_around(): - memgraph = Memgraph() - - memgraph.execute("FOREACH (i in range(1, 20000) | CREATE (:Node {id: i}));") +def test_given_supernode_when_expanding_then_expand_other_way_around(memgraph): + memgraph.execute("FOREACH (i in range(1, 1000) | CREATE (:Node {id: i}));") memgraph.execute("CREATE (:SuperNode {id: 1});") memgraph.execute("CREATE INDEX ON :SuperNode(id);") memgraph.execute("CREATE INDEX ON :SuperNode;") @@ -331,6 +329,11 @@ def test_given_supernode_when_expanding_then_expand_other_way_around(): assert expected_explain_before_analysis == result_without_analysis assert expected_explain_after_analysis == result_with_analysis + memgraph.execute("DROP INDEX ON :SuperNode") + memgraph.execute("DROP INDEX ON :Node") + memgraph.execute("DROP INDEX ON :SuperNode(id)") + memgraph.execute("DROP INDEX ON :Node(id)") + if __name__ == "__main__": sys.exit(pytest.main([__file__, "-rA"]))