test asset context instance check

dagster-io · Sep 20, 2023 · 146248f · 146248f
1 parent b376c2a
commit 146248f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/..._modules/dagster/dagster_tests/core_tests/execution_tests/test_asset_execution_context.py b/..._modules/dagster/dagster_tests/core_tests/execution_tests/test_asset_execution_context.py
@@ -129,8 +129,14 @@ def test_instance_check():
     warnings.filterwarnings("error")
 
     @asset
-    def test_instance_check(context: AssetExecutionContext):
+    def test_op_context_instance_check(context: AssetExecutionContext):
         isinstance(context, OpExecutionContext)
 
     with pytest.raises(DeprecationWarning):
-        materialize([test_instance_check])
+        materialize([test_op_context_instance_check])
+
+    @asset
+    def test_asset_context_instance_check(context: AssetExecutionContext):
+        isinstance(context, AssetExecutionContext)
+
+    materialize([test_asset_context_instance_check])