Differential D8684 Diff 40813 examples/docs_snippets/docs_snippets/concepts/io_management/test_build_io_context.py

Changeset View

Standalone View

examples/docs_snippets/docs_snippets/concepts/io_management/test_build_io_context.py

import re		import re

import pytest		import pytest
from dagster import InputContext, OutputContext, build_input_context, build_output_context, resource		from dagster import (
		InputContext,
		OutputContext,
		build_input_context,
		build_output_context,
		resource,
		)
from dagster.core.errors import DagsterInvariantViolationError		from dagster.core.errors import DagsterInvariantViolationError


def test_basic_build_input_context():		def test_basic_build_input_context():
context = build_input_context()		context = build_input_context()
assert isinstance(context, InputContext)		assert isinstance(context, InputContext)


def test_build_input_context_with_resources():		def test_build_input_context_with_resources():
@resource		@resource
def foo_def(_):		def foo_def(_):
return "bar_def"		return "bar_def"

context = build_input_context(resources={"foo": "bar", "foo_def": foo_def})		context = build_input_context(
		resources={"foo": "bar", "foo_def": foo_def}
		)
assert context.resources.foo == "bar"		assert context.resources.foo == "bar"
assert context.resources.foo_def == "bar_def"		assert context.resources.foo_def == "bar_def"


def test_build_input_context_with_cm_resource():		def test_build_input_context_with_cm_resource():
entered = []		entered = []

@resource		@resource
Show All 13 Lines	with pytest.raises(
),		),
):		):
context.resources # pylint: disable=pointless-statement		context.resources # pylint: disable=pointless-statement

del context		del context

assert entered == ["yes"]		assert entered == ["yes"]

with build_input_context(resources={"cm_resource": cm_resource}) as context:		with build_input_context(
		resources={"cm_resource": cm_resource}
		) as context:
assert context.resources.cm_resource == "foo"		assert context.resources.cm_resource == "foo"

assert entered == ["yes", "yes"]		assert entered == ["yes", "yes"]


def test_basic_build_output_context():		def test_basic_build_output_context():
context = build_output_context()		context = build_output_context()
assert isinstance(context, OutputContext)		assert isinstance(context, OutputContext)
Show All 35 Lines