1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
################################################################################
# Copyright (C) 2014 Jaakko Luttinen
#
# This file is licensed under the MIT License.
################################################################################
"""
Unit tests for `dirichlet` module.
"""
import numpy as np
from scipy import special
from bayespy.nodes import Dirichlet
from bayespy.utils import random
from bayespy.utils.misc import TestCase
class TestDirichlet(TestCase):
"""
Unit tests for Dirichlet node
"""
def test_init(self):
"""
Test the creation of Dirichlet nodes.
"""
# Some simple initializations
p = Dirichlet([1.5, 4.2, 3.5])
# Check that plates are correct
p = Dirichlet([2, 3, 4], plates=(4,3))
self.assertEqual(p.plates,
(4,3))
p = Dirichlet(np.ones((4,3,5)))
self.assertEqual(p.plates,
(4,3))
# Parent not a vector
self.assertRaises(ValueError,
Dirichlet,
4)
# Parent vector has invalid values
self.assertRaises(ValueError,
Dirichlet,
[-2,3,1])
# Plates inconsistent
self.assertRaises(ValueError,
Dirichlet,
np.ones((4,3)),
plates=(3,))
# Explicit plates too small
self.assertRaises(ValueError,
Dirichlet,
np.ones((4,3)),
plates=(1,))
pass
def test_moments(self):
"""
Test the moments of Dirichlet nodes.
"""
p = Dirichlet([2, 3, 4])
u = p._message_to_child()
self.assertAllClose(u[0],
special.psi([2,3,4]) - special.psi(2+3+4))
pass
def test_constant(self):
"""
Test the constant moments of Dirichlet nodes.
"""
p = Dirichlet([1, 1, 1])
p.initialize_from_value([0.5, 0.4, 0.1])
u = p._message_to_child()
self.assertAllClose(u[0],
np.log([0.5, 0.4, 0.1]))
pass
|
