fix tests

ecmwf · Jan 10, 2024 · d6a7f14 · d6a7f14
1 parent d21c190
commit d6a7f14
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 7 deletions.
diff --git a/tests/test_ecmwf_oper_data_fdb.py b/tests/test_ecmwf_oper_data_fdb.py
@@ -16,7 +16,7 @@ def setup_method(self, method):
             "date": {"merge": {"with": "time", "linkers": ["T", "00"]}},
             "step": {"type_change": "int"},
         }
-        self.config = {"class": "od", "expver": "0001", "levtype": "sfc", "step": 0, "type": "fc"}
+        self.config = {"class": "od", "expver": "0001", "levtype": "sfc", "stream": "oper", "type": "fc"}
         self.fdbdatacube = FDBDatacube(self.config, axis_options=self.options)
         self.slicer = HullSlicer()
         self.API = Polytope(datacube=self.fdbdatacube, engine=self.slicer, axis_options=self.options)

diff --git a/tests/test_fdb_datacube.py b/tests/test_fdb_datacube.py
@@ -20,7 +20,7 @@ def setup_method(self, method):
             "step": {"type_change": "int"},
             "number": {"type_change": "int"},
         }
-        self.config = {"class": "od", "expver": "0001", "levtype": "sfc"}
+        self.config = {"class": "od", "expver": "0001", "levtype": "sfc", "stream": "oper"}
         self.fdbdatacube = FDBDatacube(self.config, axis_options=self.options)
         self.slicer = HullSlicer()
         self.API = Polytope(datacube=self.fdbdatacube, engine=self.slicer, axis_options=self.options)
@@ -38,7 +38,6 @@ def test_fdb_datacube(self):
             Select("class", ["od"]),
             Select("stream", ["oper"]),
             Select("type", ["an"]),
-            Select("number", [1]),
             Box(["latitude", "longitude"], [0, 0], [0.2, 0.2]),
         )
         result = self.API.retrieve(request)

diff --git a/tests/test_slice_date_range_fdb.py b/tests/test_slice_date_range_fdb.py
@@ -17,7 +17,7 @@ def setup_method(self, method):
             "step": {"type_change": "int"},
             "number": {"type_change": "int"},
         }
-        self.config = {"class": "od", "expver": "0001", "levtype": "sfc", "step": "0"}
+        self.config = {"class": "od", "expver": "0001", "levtype": "sfc", "stream": "oper"}
         self.fdbdatacube = FDBDatacube(self.config, axis_options=self.options)
         self.slicer = HullSlicer()
         self.API = Polytope(datacube=self.fdbdatacube, engine=self.slicer, axis_options=self.options)
@@ -27,7 +27,6 @@ def setup_method(self, method):
     def test_fdb_datacube(self):
         request = Request(
             Select("step", [0]),
-            Select("number", [1]),
             Select("levtype", ["sfc"]),
             Span("date", pd.Timestamp("20230625T120000"), pd.Timestamp("20230626T120000")),
             Select("domain", ["g"]),
@@ -36,7 +35,6 @@ def test_fdb_datacube(self):
             Select("class", ["od"]),
             Select("stream", ["oper"]),
             Select("type", ["an"]),
-            Select("number", [1]),
             Box(["latitude", "longitude"], [0, 0], [0.2, 0.2]),
         )
         result = self.API.retrieve(request)

diff --git a/tests/test_slice_date_range_fdb_v2.py b/tests/test_slice_date_range_fdb_v2.py
@@ -16,7 +16,7 @@ def setup_method(self, method):
             "date": {"merge": {"with": "time", "linkers": ["T", "00"]}},
             "step": {"type_change": "int"},
         }
-        self.config = {"class": "ea", "expver": "0001", "levtype": "pl", "step": 0}
+        self.config = {"class": "ea", "expver": "0001", "levtype": "pl"}
         self.fdbdatacube = FDBDatacube(self.config, axis_options=self.options)
         self.slicer = HullSlicer()
         self.API = Polytope(datacube=self.fdbdatacube, engine=self.slicer, axis_options=self.options)