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