class TestNormalizer_NormalizeSample(unittest.TestCase):
def setUp(self):
self.patient = Normalizer()
def test_unit_is_unchanged_if_chemical_is_none(self):
row = self.patient.normalize_sample({
'StationId': 'ABC_WQX-abc',
'Param': None,
'Unit': 'unit',
'ResultValue': 0
})
self.assertEqual(row['Unit'], 'unit')
def test_strips_wxp(self):
row = self.patient.normalize_sample({
'StationId': 'ABC_WQX-abc',
'Param': None,
'Unit': 'unit',
'ResultValue': 0
})
self.assertEqual(row['StationId'], 'ABC-abc')
