def test(): profile = DummyData() pqc = ProfileQC(profile, saveauxiliary=False) pqc = ProfileQC(profile, saveauxiliary=True) keys = ['PRES', 'TEMP', 'PSAL', 'flag'] for v in profile.keys(): assert v in pqc.keys() assert np.allclose(profile[v], pqc[v]) for a in profile.attrs: assert a in pqc.attrs assert profile.attrs[a] == pqc.attrs[a] assert hasattr(pqc, 'flags') assert type(pqc.flags) is dict vs = list(pqc.flags.keys()) vs.remove('common') for v in vs: for f in pqc.flags[v]: assert pqc.flags[v][f].dtype == 'i1' assert hasattr(pqc, 'features') assert type(pqc.features) is dict
def test_argo(): datafile = download_testdata("20150127_prof.nc") profile = argo.profile_from_nc(datafile)[0] pqc = ProfileQC(profile, cfg='argo') print pqc.keys() print dir(pqc.input) print pqc.flags assert hasattr(pqc, 'flags') for v in ['TEMP', 'PSAL']: assert v in pqc.keys() assert len(pqc[v]) == 1034 assert v in pqc.flags for f in pqc.flags[v]: assert len(pqc.flags[v][f]) == 1034 for a in ['datetime', 'LATITUDE', 'LONGITUDE']: assert a in pqc.attributes assert type(pqc.attributes['datetime']) == datetime