def test_composed_param():
param1 = Parameter('param1', 1.)
param2 = Parameter('param2', 2.)
param3 = Parameter('param3', 3., floating=False)
param4 = Parameter(
'param4',
4.) # needed to make sure it does not only take all params as deps
a = ztf.log(3. * param1) * tf.square(param2) - param3
param_a = ComposedParameter('param_as', tensor=a)
assert isinstance(param_a.get_dependents(only_floating=True), OrderedSet)
assert param_a.get_dependents(only_floating=True) == {param1, param2}
assert param_a.get_dependents(only_floating=False) == {
param1, param2, param3
}
a_unchanged = zfit.run(a)
assert a_unchanged == zfit.run(param_a)
assert zfit.run(param2.assign(3.5))
a_changed = zfit.run(a)
assert a_changed == zfit.run(param_a)
assert a_changed != a_unchanged
with pytest.raises(LogicalUndefinedOperationError):
param_a.assign(value=5.)
with pytest.raises(LogicalUndefinedOperationError):
param_a.load(value=5., session=zfit.run.sess)
def test_composed_param():
param1 = Parameter('param1', 1.)
param2 = Parameter('param2', 2.)
param3 = Parameter('param3', 3., floating=False)
param4 = Parameter(
'param4',
4.) # needed to make sure it does not only take all params as deps
def a():
return z.log(3. * param1) * tf.square(param2) - param3
param_a = ComposedParameter('param_as',
value_fn=a,
dependents=(param1, param2, param3))
assert isinstance(param_a.get_dependents(only_floating=True), OrderedSet)
assert param_a.get_dependents(only_floating=True) == {param1, param2}
assert param_a.get_dependents(only_floating=False) == {
param1, param2, param3
}
a_unchanged = a().numpy()
assert a_unchanged == param_a.numpy()
assert param2.assign(3.5).numpy()
a_changed = a().numpy()
assert a_changed == param_a.numpy()
assert a_changed != a_unchanged
with pytest.raises(LogicalUndefinedOperationError):
param_a.assign(value=5.)
with pytest.raises(LogicalUndefinedOperationError):
param_a.assign(value=5.)