def test_names():
""" Test the regressor naming function
"""
name = 'con'
assert _regressor_names(name, 'spm') == ['con']
assert _regressor_names(name,
'spm + derivative') == ['con', 'con_derivative']
assert _regressor_names(name, 'spm + derivative + dispersion') == [
'con', 'con_derivative', 'con_dispersion'
]
assert _regressor_names(name, 'glover') == ['con']
assert _regressor_names(
name, 'glover + derivative') == ['con', 'con_derivative']
assert _regressor_names(name, 'glover + derivative + dispersion') == [
'con', 'con_derivative', 'con_dispersion'
]
def test_names():
""" Test the regressor naming function
"""
name = 'con'
assert _regressor_names(name, 'spm') == [name]
assert _regressor_names(
name, 'spm + derivative') == [name, f'{name}_derivative']
assert _regressor_names(name, 'spm + derivative + dispersion') == [
name, f'{name}_derivative', f'{name}_dispersion'
]
assert _regressor_names(name, 'glover') == [name]
assert _regressor_names(
name, 'glover + derivative') == [name, f'{name}_derivative']
assert _regressor_names(name, 'glover + derivative + dispersion') == [
name, f'{name}_derivative', f'{name}_dispersion'
]
assert _regressor_names(name, None) == [name]
assert _regressor_names(name, [None, None]) == [f"{name}_0", f"{name}_1"]
assert _regressor_names(name, "typo") == [name]
assert _regressor_names(name, ["typo", "typo"]) == \
[f"{name}_0", f"{name}_1"]
def custom_rf(tr, ov):
return np.ones(int(tr * ov))
assert _regressor_names(name, custom_rf) == \
[f"{name}_{custom_rf.__name__}"]
assert _regressor_names(name, [custom_rf]) == \
[f"{name}_{custom_rf.__name__}"]
assert _regressor_names(name, lambda tr, ov: np.ones(int(tr * ov))) == \
[f"{name}_lambda"]
assert _regressor_names(name, [lambda tr, ov: np.ones(int(tr * ov))]) == \
[f"{name}_lambda"]
with pytest.raises(ValueError,
match="Computed regressor names are not unique"):
_regressor_names(name, [
lambda tr, ov: np.ones(int(tr * ov)),
lambda tr, ov: np.ones(int(tr * ov))
])