def test_get_reparameterisation_with_class_error():
"""Test the case of class that doe not inherit from Reparameterisation.
This should raise an error.
"""
class TestReparam:
pass
with pytest.raises(TypeError) as excinfo:
get_reparameterisation(TestReparam)
assert 'Reparameterisation must be' in str(excinfo.value)
def test_get_reparameterisation_with_class():
"""Test the case of class that inherits from Reparameterisation"""
class TestReparam(Reparameterisation):
pass
reparam, kwargs = get_reparameterisation(TestReparam)
assert reparam is TestReparam
assert not kwargs
def test_get_reparameterisation_invalid_input():
"""Assert an error is raised if the input type is invalid."""
with pytest.raises(TypeError) as excinfo:
get_reparameterisation(2)
assert 'Reparameterisation must be' in str(excinfo.value)
def test_get_reparameterisation_unknown_name():
"""Assert an error is raised in the string is not recognised."""
with pytest.raises(ValueError) as excinfo:
get_reparameterisation('shift')
assert 'Unknown reparameterisation: shift' in str(excinfo.value)
def test_get_reparameterisation(inputs):
"""Test all of the included reparameterisations."""
name, expected_class, expected_kwargs = inputs
reparam, kwargs = get_reparameterisation(name)
assert reparam is expected_class
assert kwargs == expected_kwargs
def test_get_reparameterisation_defaults():
"""Test using an updated defaults dictionary."""
defaults = {'default': ('Class', {'x': 2})}
cls, kwargs = get_reparameterisation('default', defaults=defaults)
assert cls == 'Class'
assert kwargs == {'x': 2}