diff --git a/tests/helicity/test_pickle.py b/tests/helicity/test_pickle.py new file mode 100644 index 000000000..99a458ed6 --- /dev/null +++ b/tests/helicity/test_pickle.py @@ -0,0 +1,23 @@ +from __future__ import annotations + +import io +import pickle # noqa: S403 +from typing import TYPE_CHECKING + +import ampform + +if TYPE_CHECKING: + from qrules import ReactionInfo + + from ampform.helicity import HelicityModel + + +def test_model_pickling(reaction: ReactionInfo): + """See https://github.com/ComPWA/ampform/issues/471.""" + builder = ampform.get_builder(reaction) + model = builder.formulate() + stream = io.BytesIO() + pickle.dump(model, stream) + stream.seek(0) + loaded_model: HelicityModel = pickle.load(stream) # noqa: S301 + assert loaded_model.kinematic_variables == model.kinematic_variables