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')