def test_prob():
"""
Invalid probability.
"""
with pytest.raises(ValueError):
msd.Geometric([-0.1], dtype=dtype.int32)
with pytest.raises(ValueError):
msd.Geometric([1.1], dtype=dtype.int32)
def test_arguments():
"""
Args passing during initialization.
"""
g = msd.Geometric()
assert isinstance(g, msd.Distribution)
g = msd.Geometric([0.0, 0.3, 0.5, 1.0], dtype=dtype.int32)
assert isinstance(g, msd.Distribution)
def __init__(self):
super(KL, self).__init__()
self.g = msd.Geometric(0.7, dtype=dtype.int32)
def __init__(self):
super(CrossEntropy, self).__init__()
self.g = msd.Geometric(0.7, dtype=dtype.int32)
def __init__(self, shape, seed=0):
super(Sampling, self).__init__()
self.g = msd.Geometric([0.7, 0.5], seed=seed, dtype=dtype.int32)
self.shape = shape
def __init__(self):
super(Basics, self).__init__()
self.g = msd.Geometric([0.5, 0.5], dtype=dtype.int32)
def __init__(self):
super(GeometricProb1, self).__init__()
self.g = msd.Geometric(dtype=dtype.int32)
def __init__(self):
super(GeoConstruct, self).__init__()
self.g = msd.Geometric(0.5, dtype=dtype.int32)
self.g1 = msd.Geometric(dtype=dtype.int32)
def __init__(self):
super(GeometricCrossEntropy, self).__init__()
self.g1 = msd.Geometric(0.3, dtype=dtype.int32)
self.g2 = msd.Geometric(dtype=dtype.int32)
def __init__(self):
super(GeometricKl, self).__init__()
self.g1 = msd.Geometric(0.7, dtype=dtype.int32)
self.g2 = msd.Geometric(dtype=dtype.int32)
def test_seed():
with pytest.raises(TypeError):
msd.Geometric([0.1], seed='seed')
def test_name():
with pytest.raises(TypeError):
msd.Geometric([0.1], name=1.0)
def test_type():
with pytest.raises(TypeError):
msd.Geometric([0.1], dtype=dtype.float32)
