@@ -34,7 +34,7 @@ class TestQueryBuilder(tb.ModelTestCase):
3434 def test_implicit_select_01 (self ):
3535 # Schema Set
3636
37- from models .orm import default
37+ from models .orm_qb import default
3838
3939 # mypy complains if this doesn't have type annotation???
4040 users : list [default .User ] = self .client .query (default .User )
@@ -103,7 +103,7 @@ def test_implicit_select_01(self):
103103 def test_implicit_select_02 (self ):
104104 # Schema Set + filter
105105
106- from models .orm import default
106+ from models .orm_qb import default
107107
108108 users = self .client .query (default .User .filter (name = "Alice" ))
109109 self ._assertObjectsWithFields (
@@ -125,7 +125,7 @@ def test_implicit_select_02(self):
125125 def test_implicit_select_03 (self ):
126126 # Schema Set + Path
127127
128- from models .orm import default
128+ from models .orm_qb import default
129129
130130 authors = self .client .query (default .Post .author )
131131
@@ -166,7 +166,7 @@ def test_implicit_select_03(self):
166166 def test_implicit_select_04 (self ):
167167 # Schema Set + Filter + Path
168168
169- from models .orm import default
169+ from models .orm_qb import default
170170
171171 authors = self .client .query (
172172 default .Post .filter (body = "I'm Alice" ).author
@@ -191,7 +191,7 @@ def test_implicit_select_04(self):
191191 def test_implicit_select_05 (self ):
192192 # Schema Set + Path + Filter
193193
194- from models .orm import default
194+ from models .orm_qb import default
195195
196196 authors = self .client .query (default .Post .author .filter (name = "Alice" ))
197197
@@ -214,7 +214,7 @@ def test_implicit_select_05(self):
214214 def test_implicit_select_06 (self ):
215215 # Schema Set + Filter + Path + Filter
216216
217- from models .orm import default
217+ from models .orm_qb import default
218218
219219 authors = self .client .query (
220220 default .Post .filter (body = "I'm Alice" ).author .filter (name = "Alice" )
@@ -238,7 +238,7 @@ def test_implicit_select_06(self):
238238
239239 def test_implicit_select_07 (self ):
240240 # Ensure select without a schema path doesn't use a splat
241- from models .orm import default
241+ from models .orm_qb import default
242242
243243 groups = self .client .query (
244244 default .UserGroup .select (users = lambda x : x .users ).filter (
@@ -252,7 +252,7 @@ def test_implicit_select_07(self):
252252
253253 def test_implicit_select_08 (self ):
254254 # In shape: Schema Set
255- from models .orm import default
255+ from models .orm_qb import default
256256
257257 groups = self .client .query (
258258 default .UserGroup .select (users = default .User ).filter (name = "green" )
@@ -323,7 +323,7 @@ def test_implicit_select_08(self):
323323
324324 def test_implicit_select_09 (self ):
325325 # In shape: Schema Set
326- from models .orm import default
326+ from models .orm_qb import default
327327
328328 groups = self .client .query (
329329 default .UserGroup .select (
@@ -351,7 +351,7 @@ def test_implicit_select_09(self):
351351
352352 def test_implicit_select_10 (self ):
353353 # In shape: Schema Set + Path
354- from models .orm import default
354+ from models .orm_qb import default
355355
356356 groups = self .client .query (
357357 default .UserGroup .select (users = default .Post .author ).filter (
@@ -397,7 +397,7 @@ def test_implicit_select_10(self):
397397
398398 def test_implicit_select_11 (self ):
399399 # In shape: Schema Set + Filter + Path
400- from models .orm import default
400+ from models .orm_qb import default
401401
402402 groups = self .client .query (
403403 default .UserGroup .select (
@@ -425,7 +425,7 @@ def test_implicit_select_11(self):
425425
426426 def test_implicit_select_12 (self ):
427427 # In shape: Schema Set + Filter + Path
428- from models .orm import default
428+ from models .orm_qb import default
429429
430430 groups = self .client .query (
431431 default .UserGroup .select (
@@ -453,7 +453,7 @@ def test_implicit_select_12(self):
453453
454454 def test_implicit_select_13 (self ):
455455 # In shape: Schema Set + Filter + Path
456- from models .orm import default
456+ from models .orm_qb import default
457457
458458 groups = self .client .query (
459459 default .UserGroup .select (
@@ -1351,21 +1351,21 @@ def test_qb_array_agg_01(self):
13511351
13521352 def test_qb_cast_scalar_01 (self ):
13531353 # scalar to scalar
1354- from models .orm import std
1354+ from models .orm_qb import std
13551355
13561356 result = self .client .get (std .str .cast (std .int64 (1 )))
13571357 self .assertEqual (result , "1" )
13581358
13591359 def test_qb_cast_scalar_02 (self ):
13601360 # enum to scalar
1361- from models .orm import default , std
1361+ from models .orm_qb import default , std
13621362
13631363 result = self .client .get (std .str .cast (default .Color .Red ))
13641364 self .assertEqual (result , "Red" )
13651365
13661366 def test_qb_cast_scalar_03 (self ):
13671367 # scalar to enum
1368- from models .orm import default , std
1368+ from models .orm_qb import default , std
13691369
13701370 result = self .client .get (default .Color .cast (std .str ("Red" )))
13711371 self .assertEqual (result , default .Color .Red )
0 commit comments