Exemplo n.º 1
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_div(self):
     g = Gamma(shape=1, rate=2)
     h = g / 2
     self.assertEqual(repr(g), "Gamma(\nshape=\n1,\nrate=\n2\n)")
     self.assertEqual(repr(h), "Gamma(\nshape=\n1,\nrate=\n4\n)")
     h /= 2
     self.assertEqual(repr(h), "Gamma(\nshape=\n1,\nrate=\n8\n)")
Exemplo n.º 2
0
Arquivo: gamma.py Projeto: kgzy/PRML
 def test_div(self):
     g = Gamma(shape=1, rate=2)
     h = g / 2
     self.assertEqual(repr(g), "Gamma(shape=1, rate=2)")
     self.assertEqual(repr(h), "Gamma(shape=1, rate=4)")
     h /= 2
     self.assertEqual(repr(h), "Gamma(shape=1, rate=8)")
Exemplo n.º 3
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_mul(self):
     g = Gamma(shape=1, rate=2)
     h = g * 2
     self.assertEqual(repr(g), "Gamma(\nshape=\n1,\nrate=\n2\n)")
     self.assertEqual(repr(h), "Gamma(\nshape=\n1,\nrate=\n1.0\n)")
     h = 2 * h
     self.assertEqual(repr(h), "Gamma(\nshape=\n1,\nrate=\n0.5\n)")
     h *= 0.5
     self.assertEqual(repr(h), "Gamma(\nshape=\n1,\nrate=\n1.0\n)")
Exemplo n.º 4
0
Arquivo: gamma.py Projeto: kgzy/PRML
 def test_mul(self):
     g = Gamma(shape=1, rate=2)
     h = g * 2
     self.assertEqual(repr(g), "Gamma(shape=1, rate=2)")
     self.assertEqual(repr(h), "Gamma(shape=1, rate=1.0)")
     h = 2 * h
     self.assertEqual(repr(h), "Gamma(shape=1, rate=0.5)")
     h *= 0.5
     self.assertEqual(repr(h), "Gamma(shape=1, rate=1.0)")
Exemplo n.º 5
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_init(self):
     g = Gamma(shape=4, rate=6)
     self.assertEqual(g.shape_, 4)
     self.assertEqual(g.rate, 6)
Exemplo n.º 6
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_draw(self):
     g = Gamma(shape=4, rate=6)
     self.assertTrue(
         np.allclose(g.draw(10000).mean(axis=0), 2 / 3, 1e-1, 1e-1))
Exemplo n.º 7
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_pdf(self):
     g = Gamma(shape=4, rate=6)
     self.assertTrue(np.allclose(g.pdf(np.ones(1)), 0.53541047))
Exemplo n.º 8
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_var(self):
     g = Gamma(shape=4, rate=6)
     self.assertEqual(g.var, 1 / 9)
Exemplo n.º 9
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_mean(self):
     g = Gamma(shape=4, rate=6)
     self.assertEqual(g.mean, 2 / 3)
Exemplo n.º 10
0
Arquivo: gamma.py Projeto: y-ok/PRML
 def test_repr(self):
     g = Gamma(shape=1, rate=1)
     self.assertEqual(repr(g), "Gamma(\nshape=\n1,\nrate=\n1\n)")
Exemplo n.º 11
0
Arquivo: gamma.py Projeto: kgzy/PRML
 def test_repr(self):
     g = Gamma(shape=1, rate=1)
     self.assertEqual(repr(g), "Gamma(shape=1, rate=1)")