Exemple #1
0
def test_types_descent():
    @dagster_type
    class Foo(object):
        pass

    @solid(outputs=[OutputDefinition(Foo)])
    def inner_solid(_context):
        return Foo()

    middle_solid = CompositeSolidDefinition(name='middle_solid', solid_defs=[inner_solid])

    outer_solid = CompositeSolidDefinition(name='outer_solid', solid_defs=[middle_solid])

    pipe = PipelineDefinition(name='layered_types', solid_defs=[outer_solid])

    assert pipe.has_runtime_type('Foo')
Exemple #2
0
def test_builtins_available():
    pipeline = PipelineDefinition(name='test_builting_available', solids=[])
    for builtin_type in ALL_RUNTIME_BUILTINS:
        assert pipeline.has_runtime_type(builtin_type.name)
        assert pipeline.runtime_type_named(builtin_type.name).is_builtin