Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit b86d845

Browse files
Tests for Bayesnode.sample
1 parent dfa50c2 commit b86d845

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

tests/test_probability.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,6 @@ def tests():
1010
assert cpt.p(True, event) == 0.95
1111
event = {'Burglary': False, 'Earthquake': True}
1212
assert cpt.p(False, event) == 0.71
13-
# assert BoolCPT({T: 0.2, F: 0.625}).p(False, ['Burglary'], event) == 0.375
14-
# assert BoolCPT(0.75).p(False, [], {}) == 0.25
15-
# cpt = BoolCPT({True: 0.2, False: 0.7})
16-
# assert cpt.rand(['A'], {'A': True}) in [True, False]
17-
# cpt = BoolCPT({(True, True): 0.1, (True, False): 0.3,
18-
# (False, True): 0.5, (False, False): 0.7})
19-
# assert cpt.rand(['A', 'B'], {'A': True, 'B': False}) in [True, False]
2013
# #enumeration_ask('Earthquake', {}, burglary)
2114

2215
s = {'A': True, 'B': False, 'C': True, 'D': False}
@@ -87,6 +80,15 @@ def test_enumerate_joint_ask():
8780
def test_bayesnode_p():
8881
bn = BayesNode('X', 'Burglary', {T: 0.2, F: 0.625})
8982
assert bn.p(False, {'Burglary': False, 'Earthquake': True}) == 0.375
83+
assert BayesNode('W', '', 0.75).p(False, {'Random': True}) == 0.25
84+
85+
86+
def test_bayesnode_sample():
87+
X = BayesNode('X', 'Burglary', {T: 0.2, F: 0.625})
88+
assert X.sample({'Burglary': False, 'Earthquake': True}) in [True, False]
89+
Z = BayesNode('Z', 'P Q', {(True, True): 0.2, (True, False): 0.3,
90+
(False, True): 0.5, (False, False): 0.7})
91+
assert Z.sample({'P': True, 'Q': False}) in [True, False]
9092

9193

9294
def test_enumeration_ask():

0 commit comments

Comments
 (0)