def test_versioned_pickled_object_filesystem_object_manager():
with seven.TemporaryDirectory() as temp_dir:
store = VersionedPickledObjectFilesystemObjectManager(temp_dir)
context = OutputContext(
step_key="foo",
name="bar",
mapping_key=None,
metadata={},
pipeline_name="fake",
solid_def=None,
dagster_type=Any,
run_id=None,
version="version1",
)
store.handle_output(context, "cat")
assert store.has_output(context)
assert store.load_input(
InputContext(upstream_output=context,
pipeline_name="abc")) == "cat"
context_diff_version = OutputContext(
step_key="foo",
name="bar",
mapping_key=None,
metadata={},
pipeline_name="fake",
solid_def=None,
dagster_type=Any,
run_id=None,
version="version2",
)
assert not store.has_output(context_diff_version)
def test_mem_io_manager_execution():
mem_io_manager_instance = InMemoryIOManager()
output_context = OutputContext(
step_key="step_key",
name="output_name",
pipeline_name="foo",
)
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