ENH: Updating tests

ARM-DOE · Aug 21, 2024 · cf5b9a5 · cf5b9a5
1 parent 1e5081f
commit cf5b9a5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/discovery/test_improve.py b/tests/discovery/test_improve.py
@@ -2,10 +2,12 @@
 import act
 
 
-def test_get_airnow():
+def test_get_improve():
     ds = act.discovery.get_improve_data(site_id='244', start_date='1/1/2023', end_date='12/31/2023')
 
-    assert len(list(ds)) == 214
+    assert len(list(ds)) == 216
+    assert 'lat' in ds
+    assert 'lon' in ds
     assert len(ds.time.values) == 121
     assert 'aluminum_fine' in ds
     assert ds['ammonium_nitrate_fine'].values[0] == 1.41363