Make .filter() on Cases, Iterations and Realizations return a new obj…

…ect of the same class and not the base class (SearchContext).

src/fmu/sumo/explorer/objects/cases.py

@@ -13,4 +13,9 @@ def _maybe_prefetch(self, index):
 
     async def _maybe_prefetch_async(self, index):
         return
+
+    def filter(self, **kwargs):
+        sc = super().filter(**kwargs)
+        uuids = sc.uuids
+        return Cases(self, uuids)
 

src/fmu/sumo/explorer/objects/iterations.py

@@ -53,3 +53,7 @@ async def get_object_async(
 
         return obj
 
+    def filter(self, **kwargs):
+        sc = super().filter(**kwargs)
+        uuids = sc.uuids
+        return Iterations(self, uuids)

src/fmu/sumo/explorer/objects/realizations.py

@@ -53,3 +53,7 @@ async def get_object_async(
 
         return obj
 
+    def filter(self, **kwargs):
+        sc = super().filter(**kwargs)
+        uuids = sc.uuids
+        return Realizations(self, uuids)

tests/test_explorer.py

@@ -328,8 +328,8 @@ def test_seismic_case_by_uuid(explorer: Explorer, seismic_case_uuid: str):
     assert "SEGYTraceHeader" in channel_list
 
 def test_grids_and_properties(explorer: Explorer):
-    cases_with_grids = explorer.grids.cases.filter(status="keep").cases
-    cases_with_gridprops = explorer.grid_properties.cases.filter(status="keep").cases
+    cases_with_grids = explorer.grids.cases.filter(status="keep")
+    cases_with_gridprops = explorer.grid_properties.cases.filter(status="keep")
     cgs=set([case.uuid for case in cases_with_grids])
     cgps=set([case.uuid for case in cases_with_gridprops])
     assert cgs==cgps