def test_plotter_convert_to_complete_list_single_plot_no_list_allowed(
given_value, error_expected, expected_result):
if error_expected:
with pytest.raises(ValueError):
res = Plotter.convert_to_complete_list(given_value, True, 1)
else:
res = Plotter.convert_to_complete_list(given_value, True, 1)
assert res == expected_result
def test_plotter_set_parameters_function():
"""
Test setting parameters via the provided function
"""
p = Plotter(TEST_DICT)
p.set_parameters(continue_on_error=True,
**{
'B': 0,
'A': {
'test3': 'extra'
},
'D': 'not a key'
})
assert dict(p._params) == {
'A': {
'test1': 12,
'test2': 4,
'test3': 'extra'
},
'B': 0,
'C': 'title'
}
p.reset_parameters()
with pytest.raises(KeyError):
p.set_parameters(**{'B': 0, 'A': {'test3': 'extra'}, 'D': 'not a key'})
assert dict(p._params) == TEST_DICT
def test_plotter_convert_to_complete_list_multiple_plots_list(
given_value, num_plots, expected_list):
complete_list = Plotter.convert_to_complete_list(given_value, False,
num_plots)
assert complete_list == expected_list
def test_plotter_add_parameter():
"""
Test adding of custom parameters
"""
p = Plotter(TEST_DICT)
p.add_parameter('D')
assert p['D'] is None
p['D'] = 2
assert p['D'] == 2
p.add_parameter('E', default_from='C')
assert p['E'] == 'title'
p.remove_added_parameters()
p.reset_parameters()
assert dict(p._params) == TEST_DICT
def test_plotter_access():
"""
Test basic item acces and make sure the right structure was created
"""
p = Plotter(TEST_DICT)
assert dict(p._params) == TEST_DICT
#Test basic getitem method
for key, value in TEST_DICT.items():
assert p[key] == value
def test_plotter_decorator_working():
"""
Test the ensure_plotter_consistency decorator for a working function
"""
p = Plotter(TEST_DICT)
@ensure_plotter_consistency(p)
def test_function():
p.set_parameters(B=4.0)
test_function()
assert dict(p._params) == TEST_DICT
def test_plotter_decorator_add_parameter():
"""
Test the ensure_plotter_consistency decorator for a function adding custom parameters
"""
p = Plotter(TEST_DICT)
@ensure_plotter_consistency(p)
def test_function():
p.add_parameter('D')
p.add_parameter('E', default_from='C')
test_function()
assert dict(p._params) == TEST_DICT
def test_plotter_set_parameters():
"""
Test setting of parameters
"""
p = Plotter(TEST_DICT)
p['B'] = 0
assert p['B'] == 0
with pytest.raises(KeyError):
p['D'] = 'not a key'
p['A'] = {'test3': 'extra'}
assert p['A'] == {'test1': 12, 'test2': 4, 'test3': 'extra'}
with pytest.raises(ValueError):
p['A'] = 2.0
assert dict(p._params.parents) == TEST_DICT
p.reset_parameters()
assert dict(p._params) == TEST_DICT
def test_plotter_load_defaults(file_regression):
"""
Test adding of custom parameters
"""
import tempfile
p = Plotter(TEST_DICT)
p.set_defaults(continue_on_error=True,
**{
'B': 0,
'A': {
'test3': 'extra'
},
'D': 'not a key'
})
with tempfile.NamedTemporaryFile('r') as file:
p.save_defaults(file.name)
p_new = Plotter(TEST_DICT)
p_new.load_defaults(file.name)
expected_result = {
'A': {
'test1': 12,
'test2': 4,
'test3': 'extra'
},
'B': 0,
'C': 'title'
}
assert dict(p._params) == expected_result
assert dict(p._params.parents) == expected_result
def test_plotter_decorator_set_defaults():
"""
Test the ensure_plotter_consistency decorator for a function changing the defaults (raises Error)
"""
p = Plotter(TEST_DICT)
@ensure_plotter_consistency(p)
def test_function():
p.set_defaults(C='TEST')
p.set_parameters(B=4.0)
with pytest.raises(ValueError):
test_function()
assert dict(p._params) == TEST_DICT
def test_plotter_decorator_raised_error():
"""
Test the ensure_plotter_consistency decorator for a function with an error occuring durin execution
"""
p = Plotter(TEST_DICT)
@ensure_plotter_consistency(p)
def test_function():
p.set_parameters(B=4.0)
raise ValueError('Test')
with pytest.raises(ValueError, match='Test'):
test_function()
assert dict(p._params) == TEST_DICT
def test_plotter_decorator_add_parameter_raised_error():
"""
Test the ensure_plotter_consistency decorator for a function adding custom parameters with an exception occuring
"""
p = Plotter(TEST_DICT)
@ensure_plotter_consistency(p)
def test_function():
p.add_parameter('D')
p.add_parameter('E', default_from='C')
raise ValueError('Test')
with pytest.raises(ValueError, match='Test'):
test_function()
assert dict(p._params) == TEST_DICT
def test_plotter_set_defaults():
"""
Test setting of defaults
"""
p = Plotter(TEST_DICT)
p.set_defaults(continue_on_error=True,
**{
'B': 0,
'A': {
'test3': 'extra'
},
'D': 'not a key'
})
expected_result = {
'A': {
'test1': 12,
'test2': 4,
'test3': 'extra'
},
'B': 0,
'C': 'title'
}
assert dict(p._params) == expected_result
assert dict(p._params.parents) == expected_result
p.reset_defaults()
assert dict(p._params) == TEST_DICT
with pytest.raises(KeyError):
p.set_defaults(**{'B': 0, 'A': {'test3': 'extra'}, 'D': 'not a key'})
assert dict(p._params) == TEST_DICT
def test_plotter_init_defaults():
"""
Test setting defaults via initialization
"""
p = Plotter(TEST_DICT, **{
'B': 0,
'A': {
'test3': 'extra'
},
'D': 'not a key'
})
expected_result = {
'A': {
'test1': 12,
'test2': 4,
'test3': 'extra'
},
'B': 0,
'C': 'title'
}
assert dict(p._params) == expected_result
assert dict(p._params.parents) == expected_result
def test_plotter_save_defaults_complete(file_regression):
"""
Test adding of custom parameters
"""
import tempfile
p = Plotter(TEST_DICT)
p.set_defaults(continue_on_error=True,
**{
'B': 0,
'A': {
'test3': 'extra'
},
'D': 'not a key'
})
with tempfile.NamedTemporaryFile('r') as file:
p.save_defaults(file.name, save_complete=True)
txt = file.read().strip()
file_regression.check(txt)