def test_get_transformation_exception(self):
"""Test that a `KeyError` is raised if the transformation does not exist."""
mc = MeasurementChain(**self._default_init_kwargs())
mc.add_transformation(self._default_transformation())
with pytest.raises(KeyError):
mc.get_transformation("not found")
def test_get_transformation(self):
"""Test the `get_transformation` method."""
mc = MeasurementChain(**self._default_init_kwargs())
mc.add_transformation(self._default_transformation())
transformation = mc.get_transformation("transformation")
assert transformation == self._default_transformation()