diff --git a/tests/test_aggify.py b/tests/test_aggify.py index 4156ae6..c6fcda2 100644 --- a/tests/test_aggify.py +++ b/tests/test_aggify.py @@ -52,14 +52,14 @@ def test_filtering_and_projection(self): aggify.filter(age__gte=30).project(name=1, age=1) assert len(aggify.pipelines) == 2 assert aggify.pipelines[1]["$project"] == {"name": 1, "age": 1} - assert list(aggify.base_model._fields.keys()) == ["name", "age", "id"] + assert aggify.base_model._fields.keys() == {"name", "age", "id"} def test_filtering_and_projection_with_deleting_id(self): aggify = Aggify(BaseModel) aggify.filter(age__gte=30).project(name=1, age=1, id=0) assert len(aggify.pipelines) == 2 assert aggify.pipelines[1]["$project"] == {"_id": 0, "name": 1, "age": 1} - assert list(aggify.base_model._fields.keys()) == ["name", "age"] + assert aggify.base_model._fields.keys() == {"name", "age"} def test_filtering_and_ordering(self): aggify = Aggify(BaseModel) @@ -644,7 +644,7 @@ def test_project_add_new_field(self): aggify = Aggify(BaseModel) thing = list(aggify.project(test="test", id=0)) assert thing[0]["$project"] == {"test": "test", "_id": 0} - assert list(aggify.base_model._fields.keys()) == ["test"] + assert aggify.base_model._fields.keys() == {"test"} def test_lookup_raw_let(self): aggify = Aggify(BaseModel) @@ -662,7 +662,7 @@ def test_lookup_raw_let(self): "pipeline": [{"$match": {"$expr": {"$eq": ["$name", "$$test"]}}}], "as": "test_name", } - assert "test_name" in list(aggify.base_model._fields.keys()) + assert "test_name" in aggify.base_model._fields.keys() def test_group_multi_expressions(self): thing = list(Aggify(BaseModel).group(["name", "age"]))