ENH: Expanding test_airnow coverage based on latest run results (#481)

ARM-DOE · Jun 20, 2022 · 078ee8f · 078ee8f
1 parent 8e5f2a7
commit 078ee8f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/act/tests/test_discovery.py b/act/tests/test_discovery.py
@@ -107,6 +107,16 @@ def test_airnow():
         assert results['ParameterName'].values[1] == 'PM2.5'
         assert results['CategoryName'].values[0] == 'Good'
 
+        results = act.discovery.get_airnow_obs(token, zipcode=60108, distance=50)
+        assert results['ReportingArea'].values[0] == 'Chicago'
+        results = act.discovery.get_airnow_obs(token, latlon=[41.958, -88.12], distance=50)
+        assert results['StateCode'].values[0] == 'IL'
+
+        with np.testing.assert_raises(NameError):
+            results = act.discovery.get_airnow_obs(token)
+        with np.testing.assert_raises(NameError):
+            results = act.discovery.get_airnow_forecast(token, '2022-05-01')
+
         results = act.discovery.get_airnow_obs(token, date='2022-05-01', distance=50, latlon=[41.958, -88.12])
         assert results['AQI'].values[0] == 30
         assert results['ParameterName'].values[1] == 'PM2.5'