Merge pull request #21 from mahdihaghverdi/issue#18

fix issue #18
Aggify · Oct 30, 2023 · 524e613 · 524e613
2 parents 77ae54d + 2c9c19d
commit 524e613
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/aggify/aggify.py b/aggify/aggify.py
@@ -209,6 +209,7 @@ def __to_aggregate(self, query: dict[str, Any]) -> None:
                     ]
                 )
 
+    @last_out_stage_check
     def __getitem__(self, index: slice | int) -> "Aggify":
         """
         # TODO: missing docs

diff --git a/tests/test_aggify.py b/tests/test_aggify.py
@@ -358,11 +358,12 @@ def test_out_with_project_stage_error(self):
     @pytest.mark.parametrize(
         ("method", "args"),
         (
-                ("group", ("_id",)),
-                ("order_by", ("field",)),
-                ("raw", ({"$query": "query"},)),
-                ("add_fields", ({"$field": "value"},)),
-                ("filter", (Q(age=20),)),
+            ("group", ("_id",)),
+            ("order_by", ("field",)),
+            ("raw", ({"$query": "query"},)),
+            ("add_fields", ({"$field": "value"},)),
+            ("filter", (Q(age=20),)),
+            ("__getitem__", (slice(2, 10),))
         ),
     )
     def test_out_stage_error(self, method, args):
-Original file line number
+Diff line change
@@ Expand Up / @@ -209,6 +209,7 @@ def __to_aggregate(self, query: dict[str, Any]) -> None: @@
                         ]
                     )
+        @last_out_stage_check
         def __getitem__(self, index: slice | int) -> "Aggify":
             """
             # TODO: missing docs
@@ Expand Down @@