def exception_produced(self, element, name, err_value, err_type):
with self.assertRaises(TypeError) as err:
gen_name(element, name)
self.assertEqual(
str(err.exception),
"{} expected element to be 'str', found '<class '{}'>' instead"
.format(err_value, err_type))
def test_that_add_element_lines_will_call_plot_line(self, mock_plot_line):
data = {'line1': 10.0, 'line2': 20.0, 'line3': 30.0}
self.gui._add_element_lines('Cu', data)
self.assertEqual(mock_plot_line.call_count, 3)
call_list = [
mock.call(gen_name('Cu', 'line3'), 30.0, 'C0', 'Cu'),
mock.call(gen_name('Cu', 'line2'), 20.0, 'C0', 'Cu'),
mock.call(gen_name('Cu', 'line1'), 10.0, 'C0', 'Cu')
]
mock_plot_line.assert_has_calls(call_list, any_order=True)
def test_that_gen_name_with_unicode_string_does_not_throw(self):
element = u'element'
label = u'label'
result = u'element label'
self.assertEqual(gen_name(element, label), result)
def test_that_gen_name_combines_element_and_label(self):
element = 'element'
name = 'label'
self.assertEqual(gen_name(element, name), 'element label')
def test_that_gen_name_returns_name_if_it_contains_element(self):
element_name = 'name2'
name = 'name1name2name3nam4'
self.assertEqual(gen_name(element_name, name), 'name1name2name3nam4')