diff --git a/tests/convert/test_openmm_restraints.py b/tests/convert/test_openmm_restraints.py index 7f2b5ee2e..c7dcd8bf7 100644 --- a/tests/convert/test_openmm_restraints.py +++ b/tests/convert/test_openmm_restraints.py @@ -6,8 +6,12 @@ "openmm" not in sr.convert.supported_formats(), reason="openmm support is not available", ) -def test_openmm_positional_restraints(kigaki_mols, openmm_platform): - mols = kigaki_mols +@pytest.mark.parametrize("molecules", ["kigaki_mols", "merged_ethane_methanol"]) +def test_openmm_positional_restraints(molecules, openmm_platform, request): + mols = request.getfixturevalue(molecules) + + if mols[0].is_perturbable(): + mols = sr.morph.link_to_reference(mols) mol = mols[0]