Esempio n. 1
0
    def _do_expectation(expectation_context, inputs):
        check.inst_param(expectation_context, 'step_context',
                         SystemStepExecutionContext)
        value = inputs[EXPECTATION_INPUT]
        expectation_context = expectation_context.for_expectation(
            inout_def, expectation_def)
        expt_result = expectation_def.expectation_fn(
            ExpectationExecutionContext(expectation_context), value)

        if not isinstance(expt_result, ExpectationResult):
            raise DagsterInvariantViolationError((
                'Expectation for solid {solid_name} on {desc_key} {inout_name} '
                'did not return an ExpectationResult'.format(
                    solid_name=solid.name,
                    desc_key=inout_def.descriptive_key,
                    inout_name=inout_def.name,
                )))

        if expt_result.success:
            expectation_context.log.debug(
                'Expectation {key} succeeded on {value}.'.format(
                    key=expectation_context.step.key, value=value))
            yield expt_result
            yield Result(output_name=internal_output_name,
                         value=inputs[EXPECTATION_INPUT])
        else:
            expectation_context.log.debug(
                'Expectation {key} failed on {value}.'.format(
                    key=expectation_context.step.key, value=value))
            raise DagsterExpectationFailedError(expectation_context, value)
Esempio n. 2
0
 def _do_expectation(context, step, inputs):
     value = inputs[EXPECTATION_INPUT]
     info = ExpectationExecutionInfo(context, inout_def, solid, expectation_def)
     expt_result = expectation_def.expectation_fn(info, value)
     if expt_result.success:
         context.debug(
             'Expectation {key} succeeded on {value}.'.format(key=step.key, value=value)
         )
         yield Result(output_name=internal_output_name, value=inputs[EXPECTATION_INPUT])
     else:
         context.debug('Expectation {key} failed on {value}.'.format(key=step.key, value=value))
         raise DagsterExpectationFailedError(info, value)
Esempio n. 3
0
 def _do_expectation(step_context, inputs):
     check.inst_param(step_context, 'step_context', StepExecutionContext)
     value = inputs[EXPECTATION_INPUT]
     expectation_context = step_context.for_expectation(
         inout_def, expectation_def)
     expt_result = expectation_def.expectation_fn(step_context, value)
     if expt_result.success:
         expectation_context.log.debug(
             'Expectation {key} succeeded on {value}.'.format(
                 key=step_context.step.key, value=value))
         yield StepOutputValue(output_name=internal_output_name,
                               value=inputs[EXPECTATION_INPUT])
     else:
         expectation_context.log.debug(
             'Expectation {key} failed on {value}.'.format(
                 key=step_context.step.key, value=value))
         raise DagsterExpectationFailedError(expectation_context, value)