def setUp(self):
self.rule_cpd_with_rules = RuleCPD(
'A', {
('A_0', 'B_0'): 0.8,
('A_1', 'B_0'): 0.2,
('A_0', 'B_1', 'C_0'): 0.4,
('A_1', 'B_1', 'C_0'): 0.6
})
self.rule_cpd_without_rules = RuleCPD('A')
def test_init_without_errors_rules_not_none(self):
rule_cpd = RuleCPD(
'A', {
('A_0', 'B_0'): 0.8,
('A_1', 'B_0'): 0.2,
('A_0', 'B_1', 'C_0'): 0.4,
('A_1', 'B_1', 'C_0'): 0.6,
('A_0', 'B_1', 'C_1'): 0.9,
('A_1', 'B_1', 'C_1'): 0.1
})
self.assertEqual(rule_cpd.variable, 'A')
self.assertEqual(
rule_cpd.rules, {
('A_0', 'B_0'): 0.8,
('A_1', 'B_0'): 0.2,
('A_0', 'B_1', 'C_0'): 0.4,
('A_1', 'B_1', 'C_0'): 0.6,
('A_0', 'B_1', 'C_1'): 0.9,
('A_1', 'B_1', 'C_1'): 0.1
})
def test_init_without_errors_rules_none(self):
rule_cpd = RuleCPD('A')
self.assertEqual(rule_cpd.variable, 'A')
