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
