Differential D8684 Diff 40955 examples/docs_snippets/docs_snippets_tests/concepts_tests/logging_tests/test_builtin.py

Changeset View

Standalone View

examples/docs_snippets/docs_snippets_tests/concepts_tests/logging_tests/test_builtin.py

	import pytest			import pytest
	import yaml			import yaml
	from dagster import execute_pipeline			from dagster import execute_pipeline
	from dagster.utils import file_relative_path			from dagster.utils import file_relative_path
	from docs_snippets.concepts.logging.builtin_logger import demo_pipeline, demo_pipeline_error			from docs_snippets.concepts.logging.builtin_logger import (
				demo_pipeline,
				demo_pipeline_error,
				)
	from docs_snippets.concepts.logging.logging_modes import hello_modes			from docs_snippets.concepts.logging.logging_modes import hello_modes


	def test_demo_pipeline():			def test_demo_pipeline():
	assert execute_pipeline(demo_pipeline).success			assert execute_pipeline(demo_pipeline).success


	def test_demo_pipeline_config():			def test_demo_pipeline_config():
	with open(			with open(
	file_relative_path(__file__, "../../../docs_snippets/concepts/logging/config.yaml"), "r"			file_relative_path(
				__file__, "../../../docs_snippets/concepts/logging/config.yaml"
				),
				"r",
	) as fd:			) as fd:
	run_config = yaml.safe_load(fd.read())			run_config = yaml.safe_load(fd.read())
	assert execute_pipeline(demo_pipeline, run_config=run_config).success			assert execute_pipeline(demo_pipeline, run_config=run_config).success


	def test_demo_pipeline_error():			def test_demo_pipeline_error():
	with pytest.raises(Exception) as exc_info:			with pytest.raises(Exception) as exc_info:
	execute_pipeline(demo_pipeline_error)			execute_pipeline(demo_pipeline_error)
	assert str(exc_info.value) == "Somebody set up us the bomb"			assert str(exc_info.value) == "Somebody set up us the bomb"


	def test_hello_modes():			def test_hello_modes():
	assert execute_pipeline(hello_modes, mode="local").success			assert execute_pipeline(hello_modes, mode="local").success