memgraph/tests/e2e/python_query_modules_reloading/procedures/test_module.py

14 lines
440 B
Python

import mgp
from mage.test_module.test_functions import test_function as test_function1
# isort: off
# fmt: off
from mage.test_module.test_functions_dir.test_subfunctions import \
test_subfunction as test_function2
# fmt: on
@mgp.read_proc
def test(ctx: mgp.ProcCtx, a: mgp.Number, b: mgp.Number) -> mgp.Record(result1=mgp.Number, result2=mgp.Number):
return mgp.Record(result1=test_function1(a, b), result2=test_function2(a, b))