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