diff --git a/tests/test_aismod.py b/tests/test_aismod.py index 8cf1895..ce492d5 100644 --- a/tests/test_aismod.py +++ b/tests/test_aismod.py @@ -3,15 +3,15 @@ import pytest -#@pytest.mark.need_credentials -#def test_get_ais_data(): -# -# bbox = ['5.89', '62.3', '6.5', '62.7'] -# sd = '2017-01-03 08' -# ed = '2017-01-03 09' +@pytest.mark.need_credentials +def test_get_ais_data(): -# ais_ds = ais.get_AIS_data(bbox, sd, ed) -# assert len(ais_ds['time']) > 0 -# assert not all(np.isnan(v) for v in ais_ds['time']) -# assert not all(np.isnan(v) for v in ais_ds['lons']) -# assert not all(np.isnan(v) for v in ais_ds['lats']) + bbox = ['5.89', '62.3', '6.5', '62.7'] + sd = '2017-01-03 08' + ed = '2017-01-03 09' + ais_ds = ais.get_AIS_data(bbox, sd, ed) + + assert len(ais_ds['time']) > 0 + assert not all(np.isnan(v) for v in ais_ds['time']) + assert not all(np.isnan(v) for v in ais_ds['lons']) + assert not all(np.isnan(v) for v in ais_ds['lats'])