fix tests

Author: larsbuntemeyer

pyremo/preproc/core.py

@@ -169,7 +169,7 @@ def convert_units(ds):
 def check_lev(ds, invert=None):
     """Check for order of levels and invert if neccessary"""
 
-    if ds.ta.cf["vertical"].attrs.get("positive") == "down" and invert is None:
+    if ds.cf["vertical"].attrs.get("positive") == "down" and invert is None:
         invert = True
     # if "vertical" in ds.cf and auto is True:
     #    positive = ds.cf["vertical"].attrs.get("positive", None)
@@ -179,7 +179,7 @@ def check_lev(ds, invert=None):
     #    warnings.warn("could not determine positive attribute of vertical axis.")
     #    return ds
     if invert is True:
-        kwargs = {ds.ta.cf["vertical"].name: ds.ta.cf["vertical"][::-1]}
+        kwargs = {ds.cf["vertical"].name: ds.cf["vertical"][::-1]}
         print("inverting vertical axis")
         return ds.reindex(**kwargs)
     return ds

tests/test_preproc.py

@@ -2,7 +2,7 @@
 
 import pandas as pd
 import pyremo as pr
-from pyremo.preproc import gfile, remap
+from pyremo.preproc import get_gcm_dataset, remap
 from pyremo.tutorial import load_dataset, mpi_esm, mpi_esm_tos
 from pyremo.preproc.utils import get_filename
 
@@ -16,7 +16,9 @@ def ds():
 
 # @pytest.fixture
 def gcm_ds1():
-    return gfile(mpi_esm(use_cftime=True), tos=mpi_esm_tos(use_cftime=True).tos)
+    return get_gcm_dataset(
+        mpi_esm(use_cftime=True), tos=mpi_esm_tos(use_cftime=True).tos
+    )
 
 
 # @pytest.fixture
@@ -32,7 +34,7 @@ def gcm_ds2():
     }
 
     file_dict = {k: load_dataset(f).encoding["source"] for k, f in files.items()}
-    return gfile(file_dict, tos=mpi_esm_tos(use_cftime=True).tos)
+    return get_gcm_dataset(file_dict, tos=mpi_esm_tos(use_cftime=True).tos)
 
 
 # @requires_pyintorg