def test_run_method():
'''check that a notimplemented exception is raised if the run method
is called'''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
def function(arg):
'''test function'''
return 0, arg
search_method = SearchMethod(inputs, function)
with pytest.raises(NotImplementedError) as excinfo:
search_method.run()
assert "Run method should be implemented" in str(excinfo.value)
def test_set_state():
'''check that state is stored appropriately if it is set '''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
def function(arg):
'''test function'''
return 0, arg
state = "envy"
search_method = SearchMethod(inputs, function)
search_method.state = state
assert search_method.inputs is inputs
assert search_method.function is function
assert search_method.state is state
def test_invalid_function():
'''check that an exception is raised if an invalid function is provided'''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
function = "invalid"
with pytest.raises(RuntimeError) as excinfo:
_ = SearchMethod(inputs=inputs, function=function)
assert "provided function '"+function+"' is not callable" \
in str(excinfo.value)
def test_inputs_not_iterable():
'''check that an exception is raised inputs is not iteratable'''
inputs = 7
def function(arg):
'''test function'''
return 0, arg
with pytest.raises(RuntimeError) as excinfo:
_ = SearchMethod(inputs=inputs, function=function)
assert "inputs should be iterable" in str(excinfo.value)
def test_invalid_function3():
'''check that an exception is raised if too many arguments are provided
to the function'''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
def function(arg1, arg2):
'''test function'''
return arg1, arg2
with pytest.raises(RuntimeError) as excinfo:
_ = SearchMethod(inputs=inputs, function=function)
assert "provided function should have one argument" in str(excinfo.value)
def test_inputs_base_class():
'''check that an exception is raised if any of the inputs are not
subclasses of the inputs class'''
inputs = ["hello"]
def function(arg):
'''test function'''
return 0, arg
with pytest.raises(RuntimeError) as excinfo:
_ = SearchMethod(inputs, function)
assert "input should be a subclass of the Input class" \
in str(excinfo.value)
def test_init_state():
'''check that state is stored appropriately if it is
provided when the class is initialised'''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
def function(arg):
'''test function'''
return 0, arg
state = "envy"
search_method = SearchMethod(inputs=inputs, function=function, state=state)
assert search_method.inputs is inputs
assert search_method.function is function
assert search_method.state is state
def test_vanilla():
'''check that initial required values are stored appropriately if they
are provided when the class is initialised and that optional
values are set to None'''
from melody.inputs import Fixed
inputs = [Fixed(name="dummy", value="dummy")]
def function(arg):
'''test function'''
return 0, arg
search_method = SearchMethod(inputs=inputs, function=function)
assert search_method.inputs is inputs
assert search_method.function is function
assert search_method.state is None