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()