def test_station_parsing(self): from ircelsos.query_ircelsos import _process_sos_kwargs # string _, station, _ = _process_sos_kwargs('bc', station='BETR801') expected = 'BETR801' self.assertEqual(station, expected) # list of strings _, station, _ = _process_sos_kwargs('bc', station=['BETR801', 'BETR802']) expected = 'BETR801,BETR802' self.assertEqual(station, expected) # local code _, station, _ = _process_sos_kwargs('bc', station='42R801') expected = 'BETR801' self.assertEqual(station, expected) # list of local codes _, station, _ = _process_sos_kwargs('bc', station=['42R801', '42R802']) expected = 'BETR801,BETR802' self.assertEqual(station, expected)
def test_pollutant_parsing(self): from ircelsos.query_ircelsos import _process_sos_kwargs # use of short form pollutant, _, _ = _process_sos_kwargs('bc') expected = ['16111 - Black Carbon'] self.assertEqual(pollutant, expected) # long form passed through pollutant, _, _ = _process_sos_kwargs(pollutant='16111 - Black Carbon') expected = ['16111 - Black Carbon'] self.assertEqual(pollutant, expected) # raise on invalid with self.assertRaises(ValueError): pollutant, _, _ = _process_sos_kwargs('invalid') # list of strings pollutant, _, _ = _process_sos_kwargs(pollutant=['bc', 'no2']) expected = ['16111 - Black Carbon', '42602 - NO2'] self.assertEqual(pollutant, expected)