Use sushi_tset_dbt_context like other tests

Author: crericha

tests/dbt/test_model.py

@@ -694,14 +694,13 @@ def test_load_microbatch_with_ref_no_filter(
 
 @pytest.mark.slow
 def test_load_multiple_snapshots_defined_in_same_file(sushi_test_dbt_context: Context) -> None:
-    context = sushi_test_dbt_context
-    assert context.get_model("snapshots.items_snapshot")
-    assert context.get_model("snapshots.items_check_snapshot")
+    assert sushi_test_dbt_context.get_model("snapshots.items_snapshot")
+    assert sushi_test_dbt_context.get_model("snapshots.items_check_snapshot")
 
     # Make sure cache works too
-    context.load()
-    assert context.get_model("snapshots.items_snapshot")
-    assert context.get_model("snapshots.items_check_snapshot")
+    sushi_test_dbt_context.load()
+    assert sushi_test_dbt_context.get_model("snapshots.items_snapshot")
+    assert sushi_test_dbt_context.get_model("snapshots.items_check_snapshot")
 
 
 @pytest.mark.slow