Python PipelineDefinition.config_type_named Exemples

Exemple #1

def test_materialization_schema_types():
    @lambda_solid(output=OutputDefinition(types.String))
    def return_one():
        return 1

    pipeline_def = PipelineDefinition(name='test_materialization_schema_types',
                                      solids=[return_one])

    string_mat_schema = pipeline_def.config_type_named(
        'String.MaterializationSchema')

    string_json_mat_schema = string_mat_schema.fields['json'].config_type

    assert pipeline_def.config_type_named(string_json_mat_schema.name)

Exemple #2

def test_secret_field():
    @solid(config_field=Field(
        Dict({
            'password': Field(String, is_secret=True),
            'notpassword': Field(String)
        })))
    def solid_with_secret(_context):
        pass

    pipeline_def = PipelineDefinition(name='secret_pipeline',
                                      solids=[solid_with_secret])

    config_type = pipeline_def.config_type_named(
        'SecretPipeline.SolidConfig.SolidWithSecret')

    assert config_type

    password_field = config_type.fields['config'].config_type.fields[
        'password']

    assert password_field.is_secret

    notpassword_field = config_type.fields['config'].config_type.fields[
        'notpassword']

    assert not notpassword_field.is_secret