def test_chow_liu1_F(self):
bn = chow_liu(self.data)
self.assertDictEqual(
bn.F, {
0: {
'cpt': [],
'parents': [],
'values': [1, 2]
},
1: {
'cpt': [],
'parents': [0],
'values': [1, 2]
},
2: {
'cpt': [],
'parents': [0],
'values': [1, 2]
}
})
def test_chow_liu1_F(self):
bn = chow_liu(self.data)
self.assertDictEqual(bn.F,
{0: {'cpt': [], 'parents': [], 'values': [1,2]},
1: {'cpt': [], 'parents': [0], 'values': [1,2]},
2: {'cpt': [], 'parents': [0], 'values': [1,2]}})
def test_chow_liu1_E(self):
bn = chow_liu(self.data)
self.assertDictEqual(bn.E,
{0:[1,2],1:[],2:[]})
def test_chow_liu1_V(self): bn = chow_liu(self.data) self.assertListEqual(bn.V, [0,1,2])
def test_chow_liu1_E(self):
bn = chow_liu(self.data)
self.assertDictEqual(bn.E, {0: [1, 2], 1: [], 2: []})
def test_chow_liu1_V(self):
bn = chow_liu(self.data)
self.assertListEqual(bn.V, [0, 1, 2])