def filled_names(sample_names_list):
"""Return a Names class with three error codes and
three names."""
new_names = Names()
new_names.unique_error_codes(3)
new_names.lookup(sample_names_list)
return new_names
def test_unique_error_codes():
"""Test the unique_error_codes method"""
n = Names()
codes = n.unique_error_codes(3)
assert [codes[0], codes[1], codes[2]] == [0, 1, 2]
with pytest.raises(TypeError):
n.unique_error_codes(-1)
with pytest.raises(TypeError):
n.unique_error_codes("s")
def used_errors(error_list_one, error_list_two):
"""Return a names instance after five errors have been added"""
other_name = Names()
other_name.unique_error_codes(len(error_list_one))
other_name.unique_error_codes(len(error_list_two))
return other_name