def test_type():
with pytest.raises(TypeError):
msb.ScalarAffine(scale='scale')
with pytest.raises(TypeError):
msb.ScalarAffine(shift='shift')
with pytest.raises(TypeError):
msb.ScalarAffine(name=0.1)
def test_init():
"""
Test initializations.
"""
b = msb.ScalarAffine()
assert isinstance(b, msb.Bijector)
b = msb.ScalarAffine(scale=1.0)
assert isinstance(b, msb.Bijector)
b = msb.ScalarAffine(shift=2.0)
assert isinstance(b, msb.Bijector)
b = msb.ScalarAffine(3.0, 4.0)
assert isinstance(b, msb.Bijector)
def test_init():
"""
Test initializations.
"""
b = msb.Invert(msb.ScalarAffine(scale=1.0))
assert isinstance(b, msb.Bijector)
b = msb.Invert(msb.Exp())
assert isinstance(b, msb.Bijector)
def __init__(self):
super(Net3, self).__init__()
self.origin = msb.ScalarAffine(scale=2.0, shift=1.0)
self.invert = msb.Invert(self.origin)
def __init__(self):
super(BackwardJacobian, self).__init__()
self.b1 = msb.ScalarAffine(2.0, 1.0)
self.b2 = msb.ScalarAffine()
def __init__(self):
super(ForwardBackward, self).__init__()
self.b1 = msb.ScalarAffine(2.0, 1.0)
self.b2 = msb.ScalarAffine()
def __init__(self):
super(Net, self).__init__()
self.b1 = msb.ScalarAffine(1.0, 0.0)
self.b2 = msb.ScalarAffine()
def __init__(self):
super(Net3, self).__init__()
self.bijector = msb.ScalarAffine(shift=1.0, scale=2.0)
def __init__(self):
super(BackwardJacobian, self).__init__()
self.inv1 = msb.Invert(msb.Exp())
self.inv2 = msb.Invert(msb.ScalarAffine())
def __init__(self):
super(ForwardBackward, self).__init__()
self.inv1 = msb.Invert(msb.Exp())
self.inv2 = msb.Invert(msb.ScalarAffine())
def test_name():
b = msb.Invert(msb.ScalarAffine(scale=1.0))
assert b.name == 'InvertScalarAffine'
def __init__(self):
super(Net, self).__init__()
self.b1 = msb.ScalarAffine(2.0, 1.0)
self.inv = msb.Invert(self.b1)
