def test_step_cloner_should_set_steps_hyperparams_space():
p = StepClonerForEachDataInput(SomeStep())
p.set_hyperparams_space(HyperparameterSpace({
META_STEP_HP: RAND_INT_STEP_CLONER,
SOME_STEP_HP: RAND_INT_SOME_STEP
}))
assert isinstance(p.get_step().hyperparams_space, HyperparameterSpace)
assert p.get_step().hyperparams_space[SOME_STEP_HP_KEY] == RAND_INT_SOME_STEP
def test_step_cloner_should_set_steps_hyperparams():
p = StepClonerForEachDataInput(SomeStep())
p.set_hyperparams(HyperparameterSamples({
META_STEP_HP: META_STEP_HP_VALUE,
SOME_STEP_HP: SOME_STEP_HP_VALUE
}))
assert isinstance(p.hyperparams, HyperparameterSamples)
assert isinstance(p.get_step().hyperparams, HyperparameterSamples)
assert p.get_step().get_hyperparams()[SOME_STEP_HP_KEY] == SOME_STEP_HP_VALUE
def test_step_cloner_update_hyperparams_should_update_wrapped_step_hyperparams():
p = StepClonerForEachDataInput(SomeStep())
p.set_hyperparams(HyperparameterSamples({
META_STEP_HP: META_STEP_HP_VALUE,
SOME_STEP_HP: SOME_STEP_HP_VALUE
}))
p.update_hyperparams(HyperparameterSamples({
SOME_STEP_HP: SOME_STEP_HP_VALUE + 1,
}))
assert isinstance(p.hyperparams, HyperparameterSamples)
assert p.hyperparams[META_STEP_HP] == META_STEP_HP_VALUE
assert p.get_step().get_hyperparams()[SOME_STEP_HP_KEY] == SOME_STEP_HP_VALUE + 1