couple more test updates

dagster-io · Sep 21, 2023 · 0512949 · 0512949
1 parent f41150f
commit 0512949
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/examples/docs_snippets/docs_snippets/concepts/assets/asset_testing.py b/examples/docs_snippets/docs_snippets/concepts/assets/asset_testing.py
@@ -1,4 +1,4 @@
-from dagster import asset, build_op_context
+from dagster import asset, build_asset_context
 
 # start_simple_asset
 
@@ -55,7 +55,7 @@ def uses_context(context):
 
 
 def test_uses_context():
-    context = build_op_context()
+    context = build_asset_context()
     result = uses_context(context)
     assert result == "bar"
 

diff --git a/python_modules/dagster/dagster_tests/core_tests/resource_tests/test_with_resources.py b/python_modules/dagster/dagster_tests/core_tests/resource_tests/test_with_resources.py
@@ -7,7 +7,7 @@
     IOManager,
     IOManagerDefinition,
     ResourceDefinition,
-    build_op_context,
+    build_asset_context,
     execute_job,
     io_manager,
     mem_io_manager,
@@ -402,7 +402,7 @@ def test_config():
         resource_config_by_key={"foo": {"config": "blah"}, "bar": {"config": "baz"}},
     )[0]
 
-    transformed_asset(build_op_context())
+    transformed_asset(build_asset_context())
 
 
 def test_config_not_satisfied():
@@ -435,7 +435,7 @@ def test_bad_key_provided():
         },
     )[0]
 
-    transformed_asset(build_op_context())
+    transformed_asset(build_asset_context())
 
 
 def test_bad_config_provided():