From cbc9252678d0ba6a19c67f2fea7508b78c415444 Mon Sep 17 00:00:00 2001 From: JamieDeMaria Date: Wed, 20 Sep 2023 19:23:44 -0400 Subject: [PATCH] couple more test updates --- .../docs_snippets/concepts/assets/asset_testing.py | 4 ++-- .../core_tests/resource_tests/test_with_resources.py | 6 +++--- 2 files changed, 5 insertions(+), 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 index b429d913d0df2..63fc193b0f390 100644 --- 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 index 0e0570d5e20f0..62344ce0fc59f 100644 --- 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():