Skip to content

Commit

Permalink
Update test_stores_float.py
Browse files Browse the repository at this point in the history
  • Loading branch information
gmaze committed Jan 21, 2025
1 parent c9ff033 commit df09aee
Showing 1 changed file with 15 additions and 5 deletions.
20 changes: 15 additions & 5 deletions argopy/tests/test_stores_float.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import tempfile

import xarray as xr
import pandas as pd
import importlib
import shutil
import logging
Expand Down Expand Up @@ -51,6 +52,10 @@ class Test_FloatStore_Offline():
def test_init(self, wmo):
ArgoFloatOffline(wmo, host=self.host)

def test_dac_notfound(self):
with pytest.raises(ValueError):
ArgoFloatOffline(123456, host=self.host)


def id_for_host(host):
"""Get a short name for scenarios IDs, given a FTP host"""
Expand Down Expand Up @@ -137,34 +142,39 @@ def store_maker(self, request):
yield self.get_a_floatstore(wmo, host=host, cache=cache, xfail=xfail, reason=reason)

def assert_float(self, this_af):
assert isinstance(this_af.load_index(), ArgoFloatOnline)

assert hasattr(this_af, "dac")
assert isinstance(this_af.dac, str)

assert hasattr(this_af, "metadata")
assert isinstance(this_af.metadata, dict)

assert hasattr(this_af, "technicaldata")
assert isinstance(this_af.technicaldata, dict)

assert hasattr(this_af, "api_point")
assert isinstance(this_af.api_point, dict)

assert isinstance(this_af.N_CYCLES, int)
assert isinstance(this_af.path, str)
assert isinstance(this_af.host_sep, str)
assert isinstance(this_af.host_protocol, str)

assert isinstance(this_af.list_dataset(), dict)
assert is_list_of_strings(this_af.ls())

assert is_list_of_strings(this_af.lsprofiles())
assert isinstance(this_af.describe_profiles(), pd.DataFrame)

def assert_open_dataset(self, this_af):
lds = this_af.list_dataset()
log.debug(lds)
dsname, _ = random.choice(list(lds.items()))
log.info(dsname)
assert isinstance(this_af.open_dataset(dsname), xr.Dataset)

with pytest.raises(ValueError):
this_af.open_dataset('dummy_dsname')

log.debug(this_af.ls())


#########
# TESTS #
#########
Expand Down

0 comments on commit df09aee

Please sign in to comment.