class TestNormalizer_NormlizeStation(unittest.TestCase):
def setUp(self):
self.patient = Normalizer()
def test_strips_wxp(self):
row = self.patient.normalize_station({
'StationId': 'ABC_WQX-abc'
})
self.assertEqual(row['StationId'], 'ABC-abc')
def test_normalizes_station_types(self):
row = self.patient.normalize_station({
'StationId': 'ABC_WQX-abc',
'StationType': 'Cave'
})
self.assertEqual(row['StationType'], 'Other Groundwater')
