def test_mem_io_manager_execution():
mem_io_manager_instance = InMemoryIOManager()
output_context = build_output_context(step_key="step_key",
name="output_name")
mem_io_manager_instance.handle_output(output_context, 1)
input_context = build_input_context(upstream_output=output_context)
assert mem_io_manager_instance.load_input(input_context) == 1
def test_mem_io_manager_execution():
mem_io_manager_instance = InMemoryIOManager()
output_context = OutputContext(
step_key="step_key", name="output_name", pipeline_name="foo", run_id="123"
)
mem_io_manager_instance.handle_output(output_context, 1)
input_context = InputContext(
pipeline_name="foo",
upstream_output=output_context,
)
assert mem_io_manager_instance.load_input(input_context) == 1
def test_hardcoded_io_manager():
@solid
def basic_solid(_):
return 5
@pipeline(mode_defs=[
ModeDefinition(
resource_defs={
"io_manager":
IOManagerDefinition.hardcoded_io_manager(InMemoryIOManager())
})
])
def basic_pipeline():
basic_solid()
result = execute_pipeline(basic_pipeline)
assert result.success
assert result.output_for_solid("basic_solid") == 5