Differential D5721 Diff 29019 python_modules/dagster/dagster_tests/core_tests/dynamic_tests/test_dynamic_execution.py

Changeset View

Standalone View

	Show All 22 Lines
	@solid(output_defs=[DynamicOutputDefinition()])			@solid(output_defs=[DynamicOutputDefinition()])
	def emit(_):			def emit(_):
	for i in range(3):			for i in range(3):
	yield DynamicOutput(value=i, mapping_key=str(i))			yield DynamicOutput(value=i, mapping_key=str(i))


	@pipeline			@pipeline
	def dynamic_pipeline():			def dynamic_pipeline():
	multiply_by_two(multiply_inputs(emit(), emit_ten()))			numbers = emit()
				# pylint: disable=no-member
				numbers.map(lambda num: multiply_by_two(multiply_inputs(num, emit_ten())))


	def test_map():			def test_map():
	result = execute_pipeline(dynamic_pipeline)			result = execute_pipeline(dynamic_pipeline)
	assert result.success			assert result.success
	assert result.result_for_solid("multiply_inputs").output_value() == {"0": 0, "1": 10, "2": 20}			assert result.result_for_solid("multiply_inputs").output_value() == {"0": 0, "1": 10, "2": 20}
	assert result.result_for_solid("multiply_by_two").output_value() == {"0": 0, "1": 20, "2": 40}			assert result.result_for_solid("multiply_by_two").output_value() == {"0": 0, "1": 20, "2": 40}

	▲ Show 20 Lines • Show All 43 Lines • Show Last 20 Lines