memgraph/tests/benchmark_infra/groups/aggregation/setup.py