From 59d9b94426a90fd9ba641f252e833eede0f1c0f4 Mon Sep 17 00:00:00 2001 From: Hendrik Roch Date: Wed, 7 Aug 2024 11:32:50 -0400 Subject: [PATCH 1/2] Add ABC FlowInterface tests --- tests/test_FlowInterface.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/test_FlowInterface.py diff --git a/tests/test_FlowInterface.py b/tests/test_FlowInterface.py new file mode 100644 index 0000000..f8044e8 --- /dev/null +++ b/tests/test_FlowInterface.py @@ -0,0 +1,32 @@ +# =================================================== +# +# Copyright (c) 2023-2024 +# SPARKX Team +# +# GNU General Public License (GPLv3 or later) +# +# =================================================== + +from sparkx.flow.FlowInterface import FlowInterface + +class DummyFlow(FlowInterface): + def __init__(self, *args, **kwargs): + self.args = args + self.kwargs = kwargs + + def integrated_flow(self, particle_data, *args, **kwargs): + return "integrated_flow called" + + def differential_flow(self, particle_data, bins, flow_as_function_of, *args, **kwargs): + return "differential_flow called" + + +def test_integrated_flow(): + flow = DummyFlow() + result = flow.integrated_flow("particle_data") + assert result == "integrated_flow called" + +def test_differential_flow(): + flow = DummyFlow() + result = flow.differential_flow("particle_data", "bins", "flow_as_function_of") + assert result == "differential_flow called" \ No newline at end of file From ba73173ad148653c34eb4fd6fe9e37c538398aed Mon Sep 17 00:00:00 2001 From: Hendrik Roch Date: Wed, 7 Aug 2024 11:42:39 -0400 Subject: [PATCH 2/2] Format FlowInterface test --- tests/test_FlowInterface.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/test_FlowInterface.py b/tests/test_FlowInterface.py index f8044e8..033c349 100644 --- a/tests/test_FlowInterface.py +++ b/tests/test_FlowInterface.py @@ -9,6 +9,7 @@ from sparkx.flow.FlowInterface import FlowInterface + class DummyFlow(FlowInterface): def __init__(self, *args, **kwargs): self.args = args @@ -17,16 +18,21 @@ def __init__(self, *args, **kwargs): def integrated_flow(self, particle_data, *args, **kwargs): return "integrated_flow called" - def differential_flow(self, particle_data, bins, flow_as_function_of, *args, **kwargs): + def differential_flow( + self, particle_data, bins, flow_as_function_of, *args, **kwargs + ): return "differential_flow called" - + def test_integrated_flow(): flow = DummyFlow() result = flow.integrated_flow("particle_data") assert result == "integrated_flow called" + def test_differential_flow(): flow = DummyFlow() - result = flow.differential_flow("particle_data", "bins", "flow_as_function_of") - assert result == "differential_flow called" \ No newline at end of file + result = flow.differential_flow( + "particle_data", "bins", "flow_as_function_of" + ) + assert result == "differential_flow called"