@@ -64,7 +64,6 @@ def test_conversion_various_types(self):
6464 def test_conversion_various_array_types (self ):
6565 self ._test_conversion_various_types (self ._test_conversion_valid_array_type )
6666
67-
6867 def test_conversion_getfield (self ):
6968 expr = {"$eq" : [{"$getField" : {"input" : "$item" , "field" : "age" }}, 10 ]}
7069 self .assertConversionEqual (expr , {"item.age" : 10 })
@@ -83,6 +82,7 @@ def test_conversion_nested_getfield(self):
8382 }
8483 self .assertConversionEqual (expr , {"item.shel_life.age" : 10 })
8584
85+
8686class InTests (ConversionTestCase ):
8787 def test_conversion (self ):
8888 expr = {"$in" : ["$category" , ["electronics" , "books" , "clothing" ]]}
@@ -102,6 +102,7 @@ def test_conversion_various_types(self):
102102 for _type , val in self .CONVERTIBLE_TYPES .items ():
103103 with self .subTest (_type = _type , val = val ):
104104 self ._test_conversion_valid_type (val )
105+
105106 def test_conversion_getfield (self ):
106107 expr = {"$in" : [{"$getField" : {"input" : "$item" , "field" : "age" }}, [10 ]]}
107108 expected = {"item.age" : {"$in" : [10 ]}}
@@ -130,12 +131,14 @@ def test_conversion_dual_getfield_ineligible(self):
130131 "field" : "age" ,
131132 }
132133 },
133- [{
134- "$getField" : {
135- "input" : "$value" ,
136- "field" : "age" ,
134+ [
135+ {
136+ "$getField" : {
137+ "input" : "$value" ,
138+ "field" : "age" ,
139+ }
137140 }
138- } ],
141+ ],
139142 ]
140143 }
141144 self .assertNotOptimizable (expr )
@@ -201,7 +204,7 @@ def test_mixed(self):
201204 {"$in" : ["$category" , ["electronics" , "books" ]]},
202205 {"$eq" : ["$verified" , True ]},
203206 {"$lte" : ["$price" , 2000 ]},
204- {"$eq" : [{"$getField" : {"input" : "$root" , "field" : "age" }}, 10 ]}
207+ {"$eq" : [{"$getField" : {"input" : "$root" , "field" : "age" }}, 10 ]},
205208 ]
206209 }
207210 expected = {
@@ -210,7 +213,7 @@ def test_mixed(self):
210213 {"category" : {"$in" : ["electronics" , "books" ]}},
211214 {"verified" : True },
212215 {"price" : {"$lte" : 2000 }},
213- {"root.age" : 10 }
216+ {"root.age" : 10 },
214217 ]
215218 }
216219 self .assertConversionEqual (expr , expected )
0 commit comments