Differential D8684 Diff 40955 examples/docs_snippets/docs_snippets_tests/concepts_tests/configuration_tests/test_configured_example.py
Changeset View
Changeset View
Standalone View
Standalone View
examples/docs_snippets/docs_snippets_tests/concepts_tests/configuration_tests/test_configured_example.py
| import yaml | import yaml | ||||
| from dagster import ModeDefinition, execute_pipeline, pipeline, solid | from dagster import ModeDefinition, execute_pipeline, pipeline, solid | ||||
| from dagster.utils import file_relative_path | from dagster.utils import file_relative_path | ||||
| from docs_snippets.concepts.configuration.config_map_example import unsigned_s3_session | from docs_snippets.concepts.configuration.config_map_example import ( | ||||
| unsigned_s3_session, | |||||
| ) | |||||
| from docs_snippets.concepts.configuration.configured_example import ( | from docs_snippets.concepts.configuration.configured_example import ( | ||||
| east_unsigned_s3_session, | east_unsigned_s3_session, | ||||
| s3_session, | s3_session, | ||||
| west_signed_s3_session, | west_signed_s3_session, | ||||
| west_unsigned_s3_session, | west_unsigned_s3_session, | ||||
| ) | ) | ||||
| def test_config_map_example(): | def test_config_map_example(): | ||||
| execute_pipeline_with_resource_def( | execute_pipeline_with_resource_def( | ||||
| unsigned_s3_session, run_config={"resources": {"key": {"config": {"region": "us-east-1"}}}} | unsigned_s3_session, | ||||
| run_config={ | |||||
| "resources": {"key": {"config": {"region": "us-east-1"}}} | |||||
| }, | |||||
| ) | ) | ||||
| def execute_pipeline_with_resource_def(resource_def, run_config=None): | def execute_pipeline_with_resource_def(resource_def, run_config=None): | ||||
| @solid(required_resource_keys={"key"}) | @solid(required_resource_keys={"key"}) | ||||
| def a_solid(): | def a_solid(): | ||||
| pass | pass | ||||
| Show All 9 Lines | def test_configured_example(): | ||||
| execute_pipeline_with_resource_def(east_unsigned_s3_session) | execute_pipeline_with_resource_def(east_unsigned_s3_session) | ||||
| execute_pipeline_with_resource_def(west_unsigned_s3_session) | execute_pipeline_with_resource_def(west_unsigned_s3_session) | ||||
| execute_pipeline_with_resource_def(west_signed_s3_session) | execute_pipeline_with_resource_def(west_signed_s3_session) | ||||
| def test_configured_example_yaml(): | def test_configured_example_yaml(): | ||||
| with open( | with open( | ||||
| file_relative_path( | file_relative_path( | ||||
| __file__, "../../../docs_snippets/concepts/configuration/configured_example.yaml" | __file__, | ||||
| "../../../docs_snippets/concepts/configuration/configured_example.yaml", | |||||
| ), | ), | ||||
| "r", | "r", | ||||
| ) as fd: | ) as fd: | ||||
| run_config = yaml.safe_load(fd.read()) | run_config = yaml.safe_load(fd.read()) | ||||
| execute_pipeline_with_resource_def(s3_session, run_config) | execute_pipeline_with_resource_def(s3_session, run_config) | ||||