def setUp(self): self.conversion_map = { 'a': INT_TRANSF, 'b': STR_TRANSF, 'c': FLOAT_TRANSF} self.converter = Converter(conversion_map=self.conversion_map) self.maxDiff = None
class ConvertTestCase(unittest.TestCase): def setUp(self): self.conversion_map = {'a': INT_TRANSF, 'b': STR_TRANSF, 'c': FLOAT_TRANSF} self.converter = Converter(conversion_map=self.conversion_map) self.maxDiff = None def test_conversion_correct_values_for_keys(self): entry = {'a': '45', 'b': ' hazard'} exp_entry = {'a': 45, 'b': 'hazard'} self.assertEqual(exp_entry, self.converter.convert(entry)) def test_conversion_incorrect_values_for_keys(self): entry = {'a': '45.78', 'b': 'risk8'} exp_entry = {'a': None, 'b': 'risk8'} self.assertEqual(exp_entry, self.converter.convert(entry)) def test_conversion_float_and_empty_value(self): entry = {'b': '', 'c': '45.90'} exp_entry = {'b': None, 'c': 45.90} self.assertEqual(exp_entry, self.converter.convert(entry))