def test_opt_nullable_dict_param(): assert check.opt_nullable_dict_param(None, "opt_nullable_dict_param") is None assert check.opt_nullable_dict_param({}, "opt_nullable_dict_param") == {} assert check.opt_nullable_dict_param(frozendict(), "opt_nullable_dict_param") == {} ddict = {"a": 2} assert check.opt_nullable_dict_param(ddict, "opt_nullable_dict_param") == ddict class Foo(object): pass class Bar(Foo): pass ddict_class = {"a": Bar} assert (check.opt_nullable_dict_param(ddict_class, "opt_nullable_dict_param", value_class=Foo) == ddict_class) with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param(1, "opt_nullable_dict_param") with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param("foo", "opt_nullable_dict_param")
def test_opt_nullable_dict_param(): assert check.opt_nullable_dict_param(None, 'opt_nullable_dict_param') is None assert check.opt_nullable_dict_param({}, 'opt_nullable_dict_param') == {} assert check.opt_nullable_dict_param(frozendict(), 'opt_nullable_dict_param') == {} ddict = {'a': 2} assert check.opt_nullable_dict_param(ddict, 'opt_nullable_dict_param') == ddict class Foo(object): pass class Bar(Foo): pass ddict_class = {'a': Bar} assert (check.opt_nullable_dict_param(ddict_class, 'opt_nullable_dict_param', value_class=Foo) == ddict_class) with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param(1, 'opt_nullable_dict_param') with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param('foo', 'opt_nullable_dict_param')
def with_additional_config(self, run_config): '''Return a new PresetDefinition with additional config merged in to the existing config.''' check.opt_nullable_dict_param(run_config, 'run_config') if run_config is None: return self else: return PresetDefinition( name=self.name, solid_selection=self.solid_selection, mode=self.mode, run_config=merge_dicts(self.run_config, run_config), )
def with_additional_config(self, environment_dict): '''return a new PresetDefinition with additional config merged in to the existing config''' check.opt_nullable_dict_param(environment_dict, 'environment_dict') if environment_dict is None: return self else: return PresetDefinition( name=self.name, solid_subset=self.solid_subset, mode=self.mode, environment_dict=merge_dicts(self.environment_dict, environment_dict), )
def with_additional_config(self, run_config): """Return a new PresetDefinition with additional config merged in to the existing config.""" check.opt_nullable_dict_param(run_config, "run_config") if run_config is None: return self else: return PresetDefinition( name=self.name, solid_selection=self.solid_selection, mode=self.mode, tags=self.tags, run_config=deep_merge_dicts(self.run_config, run_config), )
def __new__( cls, key: str, inputs: List["ExecutionStepInputSnap"], outputs: List["ExecutionStepOutputSnap"], solid_handle_id: str, kind: StepKind, metadata_items: List["ExecutionPlanMetadataItemSnap"], tags: Optional[Dict[str, str]] = None, step_handle: Optional[StepHandleUnion] = None, ): return super(ExecutionStepSnap, cls).__new__( cls, key=check.str_param(key, "key"), inputs=check.list_param(inputs, "inputs", ExecutionStepInputSnap), outputs=check.list_param(outputs, "outputs", ExecutionStepOutputSnap), solid_handle_id=check.str_param(solid_handle_id, "solid_handle_id"), kind=check.inst_param(kind, "kind", StepKind), metadata_items=check.list_param(metadata_items, "metadata_items", ExecutionPlanMetadataItemSnap), tags=check.opt_nullable_dict_param(tags, "tags", key_type=str, value_type=str), step_handle=check.opt_inst_param(step_handle, "step_handle", StepHandleTypes), )
def __new__( cls, key, inputs, outputs, solid_handle_id, kind, metadata_items, tags=None, step_handle=None, ): return super(ExecutionStepSnap, cls).__new__( cls, key=check.str_param(key, "key"), inputs=check.list_param(inputs, "inputs", ExecutionStepInputSnap), outputs=check.list_param(outputs, "outputs", ExecutionStepOutputSnap), solid_handle_id=check.str_param(solid_handle_id, "solid_handle_id"), kind=check.inst_param(kind, "kind", StepKind), metadata_items=check.list_param(metadata_items, "metadata_items", ExecutionPlanMetadataItemSnap), tags=check.opt_nullable_dict_param(tags, "tags", key_type=str), step_handle=check.opt_inst_param(step_handle, "step_handle", StepHandleTypes), )
def test_opt_nullable_dict_param(): assert check.opt_nullable_dict_param(None, 'opt_nullable_dict_param') is None assert check.opt_nullable_dict_param({}, 'opt_nullable_dict_param') == {} ddict = {'a': 2} assert check.opt_nullable_dict_param(ddict, 'opt_nullable_dict_param') == ddict with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param(1, 'opt_nullable_dict_param') with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param('foo', 'opt_nullable_dict_param')
def test_opt_nullable_dict_param(): assert check.opt_nullable_dict_param(None, "opt_nullable_dict_param") is None assert check.opt_nullable_dict_param({}, "opt_nullable_dict_param") == {} assert check.opt_nullable_dict_param(frozendict(), "opt_nullable_dict_param") == {} ddict = {"a": 2} assert check.opt_nullable_dict_param(ddict, "opt_nullable_dict_param") == ddict with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param(1, "opt_nullable_dict_param") with pytest.raises(ParameterCheckError): check.opt_nullable_dict_param("foo", "opt_nullable_dict_param")
def __init__( self, name: Optional[str] = None, input_defs: Optional[Sequence[InputDefinition]] = None, output_defs: Optional[Sequence[OutputDefinition]] = None, description: Optional[str] = None, required_resource_keys: Optional[Set[str]] = None, config_schema: Optional[Union[Any, Dict[str, Any]]] = None, tags: Optional[Dict[str, Any]] = None, version: Optional[str] = None, decorator_takes_context: Optional[bool] = True, retry_policy: Optional[RetryPolicy] = None, ins: Optional[Dict[str, In]] = None, out: Optional[Union[Out, Dict[str, Out]]] = None, ): self.name = check.opt_str_param(name, "name") self.input_defs = check.opt_nullable_sequence_param( input_defs, "input_defs", of_type=InputDefinition) self.output_defs = output_defs self.decorator_takes_context = check.bool_param( decorator_takes_context, "decorator_takes_context") self.description = check.opt_str_param(description, "description") # these will be checked within SolidDefinition self.required_resource_keys = required_resource_keys self.tags = tags self.version = version self.retry_policy = retry_policy # config will be checked within SolidDefinition self.config_schema = config_schema self.ins = check.opt_nullable_dict_param(ins, "ins", key_type=str, value_type=In) self.out = out