From cf5b9a580de7f54ec32714274c57713cfc1b55c0 Mon Sep 17 00:00:00 2001 From: AdamTheisen Date: Wed, 21 Aug 2024 10:00:45 -0500 Subject: [PATCH] ENH: Updating tests --- tests/discovery/test_improve.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/discovery/test_improve.py b/tests/discovery/test_improve.py index 1bc31025ba..c8a75f1fd6 100644 --- 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