@@ -243,7 +243,7 @@ def alias_query_data(alias_table):
243243 AliasKeyModel .delete (datum ["name" ])
244244
245245
246- @pytest .fixture ( scope = "module" )
246+ @pytest .fixture
247247def nested_query_data (nested_table ):
248248 presets = [dict ()] * 5
249249 data = [datum for datum in [nested_model_data_generator (** i ) for i in presets ]]
@@ -257,7 +257,7 @@ def nested_query_data(nested_table):
257257 NestedModel .delete ((datum [NestedModel .Config .hash_key ], datum [NestedModel .Config .range_key ]))
258258
259259
260- @pytest .fixture ( scope = "module" )
260+ @pytest .fixture
261261def nested_query_data_empty_ticket (nested_table ):
262262 presets = [dict ()] * 5
263263 data = [datum for datum in [nested_model_data_generator (include_ticket = False , ** i ) for i in presets ]]
@@ -460,14 +460,15 @@ def test_query_scan_complex(dynamo, complex_query_data):
460460
461461def test_query_with_nested_model (dynamo , nested_query_data ):
462462 res = NestedModel .query ()
463- res_data = [m .ticket for m in res ]
464- assert any (elem is not None for elem in res_data )
463+ for m in res :
464+ assert m .ticket .created_time is not None
465+ assert m .ticket .number is not None
466+ assert type (m .ticket ) is Ticket
465467
466468
467469def test_query_with_nested_model_optional (dynamo , nested_query_data_empty_ticket ):
468470 res = NestedModel .query ()
469- res_data = [m .ticket for m in res ]
470- assert any (elem is None for elem in res_data )
471+ assert all ([m .ticket is None for m in res ])
471472
472473
473474def test_query_alias_save (dynamo ):
0 commit comments