def test_two_downstream_assets_job(): @asset def asset1(): return 1 @asset def asset2_a(asset1): assert asset1 == 1 @asset def asset2_b(asset1): assert asset1 == 1 assets_job = build_assets_job("assets_job", [asset1, asset2_a, asset2_b]) external_asset_nodes = external_asset_graph_from_defs([assets_job], source_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("asset1"), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey("asset2_a"), input_name="asset1" ), ExternalAssetDependedBy( downstream_asset_key=AssetKey("asset2_b"), input_name="asset1" ), ], op_name="asset1", op_description=None, job_names=["assets_job"], output_name="result", output_description=None, ), ExternalAssetNode( asset_key=AssetKey("asset2_a"), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey("asset1"), input_name="asset1") ], depended_by=[], op_name="asset2_a", op_description=None, job_names=["assets_job"], output_name="result", output_description=None, ), ExternalAssetNode( asset_key=AssetKey("asset2_b"), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey("asset1"), input_name="asset1") ], depended_by=[], op_name="asset2_b", op_description=None, job_names=["assets_job"], output_name="result", output_description=None, ), ]
def test_input_name_or_output_name_dep_by(): with pytest.raises(CheckError, match="input `foo` and output `bar`"): ExternalAssetDependedBy( downstream_asset_key=AssetKey("bar"), input_name="foo", output_name="bar" ) with pytest.raises(CheckError, match="input `None` and output `None`"): ExternalAssetDependedBy( downstream_asset_key=AssetKey("bar"), input_name=None, output_name=None )
def test_input_name_matches_output_name(): not_result = SourceAsset(key=AssetKey("not_result"), description=None) @asset(ins={"result": AssetIn(asset_key=AssetKey("not_result"))}) def something(result): # pylint: disable=unused-argument pass assets_job = build_assets_job("assets_job", [something], source_assets=[not_result]) external_asset_nodes = external_asset_graph_from_defs([assets_job], source_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("not_result"), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey("something"), input_name="result" ) ], job_names=[], ), ExternalAssetNode( asset_key=AssetKey("something"), dependencies=[ ExternalAssetDependency( upstream_asset_key=AssetKey("not_result"), input_name="result" ) ], depended_by=[], op_name="something", output_name="result", job_names=["assets_job"], ), ]
def test_used_source_asset(): bar = SourceAsset(key=AssetKey("bar"), description="def") @asset def foo(bar): assert bar job1 = build_assets_job("job1", [foo], source_assets=[bar]) external_asset_nodes = external_asset_graph_from_defs( [job1], source_assets_by_key={AssetKey("bar"): bar} ) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("bar"), op_description="def", dependencies=[], depended_by=[ ExternalAssetDependedBy(downstream_asset_key=AssetKey(["foo"]), input_name="bar") ], job_names=[], ), ExternalAssetNode( asset_key=AssetKey("foo"), op_name="foo", op_description=None, dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey(["bar"]), input_name="bar") ], depended_by=[], job_names=["job1"], output_name="result", output_description=None, ), ]
def test_source_asset_with_op(): foo = SourceAsset(key=AssetKey("foo"), description=None) @asset def bar(foo): # pylint: disable=unused-argument pass assets_job = build_assets_job("assets_job", [bar], source_assets=[foo]) external_asset_nodes = external_asset_graph_from_defs([assets_job], source_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("foo"), op_description=None, dependencies=[], depended_by=[ExternalAssetDependedBy(AssetKey("bar"), input_name="foo")], job_names=[], ), ExternalAssetNode( asset_key=AssetKey("bar"), op_name="bar", op_description=None, dependencies=[ExternalAssetDependency(AssetKey("foo"), input_name="foo")], depended_by=[], job_names=["assets_job"], output_name="result", ), ]
def test_cross_pipeline_asset_dependency(): @asset def asset1(): return 1 @asset def asset2(asset1): assert asset1 == 1 @pipeline def asset1_graph(): asset1() @pipeline def asset2_graph(): asset2() # pylint: disable=no-value-for-parameter external_asset_nodes = external_asset_graph_from_defs( [asset1_graph, asset2_graph], foreign_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("asset1"), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey("asset2"), input_name="asset1") ], op_name="asset1", op_description=None, job_names=["asset1_graph"], ), ExternalAssetNode( asset_key=AssetKey("asset2"), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey("asset1"), input_name="asset1") ], depended_by=[], op_name="asset2", op_description=None, job_names=["asset2_graph"], ), ]
def test_cross_job_asset_dependency(): @asset def asset1(): return 1 @asset def asset2(asset1): assert asset1 == 1 assets_job1 = build_assets_job("assets_job1", [asset1]) assets_job2 = build_assets_job("assets_job2", [asset2], source_assets=[asset1]) external_asset_nodes = external_asset_graph_from_defs( [assets_job1, assets_job2], source_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("asset1"), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey("asset2"), input_name="asset1") ], op_name="asset1", op_description=None, job_names=["assets_job1"], output_name="result", output_description=None, ), ExternalAssetNode( asset_key=AssetKey("asset2"), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey("asset1"), input_name="asset1") ], depended_by=[], op_name="asset2", op_description=None, job_names=["assets_job2"], output_name="result", output_description=None, ), ]
def test_two_asset_pipeline(): @asset def asset1(): return 1 @asset def asset2(asset1): assert asset1 == 1 @pipeline def my_graph(): asset2(asset1()) external_asset_nodes = external_asset_graph_from_defs( [my_graph], foreign_assets_by_key={}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("asset1"), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey("asset2"), input_name="asset1") ], op_name="asset1", op_description=None, job_names=["my_graph"], ), ExternalAssetNode( asset_key=AssetKey("asset2"), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey("asset1"), input_name="asset1") ], depended_by=[], op_name="asset2", op_description=None, job_names=["my_graph"], ), ]
def test_used_foreign_asset(): bar = ForeignAsset(key=AssetKey("bar"), description="def") @asset def foo(bar): assert bar @job def job1(): foo() # pylint: disable=no-value-for-parameter external_asset_nodes = external_asset_graph_from_defs( [job1], foreign_assets_by_key={AssetKey("bar"): bar}) assert external_asset_nodes == [ ExternalAssetNode( asset_key=AssetKey("bar"), op_description="def", dependencies=[], depended_by=[ ExternalAssetDependedBy(downstream_asset_key=AssetKey(["foo"]), input_name="bar") ], job_names=[], ), ExternalAssetNode( asset_key=AssetKey("foo"), op_name="foo", op_description=None, dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey(["bar"]), input_name="bar") ], depended_by=[], job_names=["job1"], ), ]
def test_inter_op_dependency(): @asset def in1(): pass @asset def in2(): pass @asset def downstream(only_in, mixed, only_out): # pylint: disable=unused-argument pass @multi_asset( outs={"only_in": Out(), "mixed": Out(), "only_out": Out()}, internal_asset_deps={ "mixed": {AssetKey("in1"), AssetKey("only_in")}, "only_out": {AssetKey("only_in"), AssetKey("mixed")}, }, ) def assets(in1, in2): # pylint: disable=unused-argument pass assets_job = build_assets_job("assets_job", [in1, in2, assets, downstream]) external_asset_nodes = external_asset_graph_from_defs([assets_job], source_assets_by_key={}) # sort so that test is deterministic sorted_nodes = sorted( [ node._replace( dependencies=sorted(node.dependencies, key=lambda d: d.upstream_asset_key), depended_by=sorted(node.depended_by, key=lambda d: d.downstream_asset_key), ) for node in external_asset_nodes ], key=lambda n: n.asset_key, ) assert sorted_nodes == [ ExternalAssetNode( asset_key=AssetKey(["downstream"]), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey(["mixed"]), input_name="mixed"), ExternalAssetDependency( upstream_asset_key=AssetKey(["only_in"]), input_name="only_in" ), ExternalAssetDependency( upstream_asset_key=AssetKey(["only_out"]), input_name="only_out" ), ], depended_by=[], op_name="downstream", op_description=None, job_names=["assets_job"], output_name="result", metadata_entries=[], ), ExternalAssetNode( asset_key=AssetKey(["in1"]), dependencies=[], depended_by=[ ExternalAssetDependedBy(downstream_asset_key=AssetKey(["mixed"]), input_name="in1"), ExternalAssetDependedBy( downstream_asset_key=AssetKey(["only_in"]), input_name="in1" ), ], op_name="in1", op_description=None, job_names=["assets_job"], output_name="result", metadata_entries=[], ), ExternalAssetNode( asset_key=AssetKey(["in2"]), dependencies=[], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey(["only_in"]), input_name="in2" ) ], op_name="in2", op_description=None, job_names=["assets_job"], output_name="result", metadata_entries=[], ), ExternalAssetNode( asset_key=AssetKey(["mixed"]), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey(["in1"]), input_name="in1"), ExternalAssetDependency( upstream_asset_key=AssetKey(["only_in"]), output_name="only_in" ), ], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey(["downstream"]), input_name="mixed" ), ExternalAssetDependedBy( downstream_asset_key=AssetKey(["only_out"]), output_name="mixed" ), ], op_name="assets", op_description=None, job_names=["assets_job"], output_name="mixed", metadata_entries=[ MetadataEntry( label=".dagster/asset_deps", description=None, entry_data=MetadataValue.text("[set] (unserializable)"), ) ], ), ExternalAssetNode( asset_key=AssetKey(["only_in"]), dependencies=[ ExternalAssetDependency(upstream_asset_key=AssetKey(["in1"]), input_name="in1"), ExternalAssetDependency(upstream_asset_key=AssetKey(["in2"]), input_name="in2"), ], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey(["downstream"]), input_name="only_in" ), ExternalAssetDependedBy( downstream_asset_key=AssetKey(["mixed"]), output_name="only_in" ), ExternalAssetDependedBy( downstream_asset_key=AssetKey(["only_out"]), output_name="only_in" ), ], op_name="assets", op_description=None, job_names=["assets_job"], output_name="only_in", metadata_entries=[], ), ExternalAssetNode( asset_key=AssetKey(["only_out"]), dependencies=[ ExternalAssetDependency( upstream_asset_key=AssetKey(["mixed"]), output_name="mixed" ), ExternalAssetDependency( upstream_asset_key=AssetKey(["only_in"]), output_name="only_in" ), ], depended_by=[ ExternalAssetDependedBy( downstream_asset_key=AssetKey(["downstream"]), input_name="only_out" ), ], op_name="assets", op_description=None, job_names=["assets_job"], output_name="only_out", metadata_entries=[ MetadataEntry( label=".dagster/asset_deps", description=None, entry_data=MetadataValue.text("[set] (unserializable)"), ) ], ), ]