Tests for Bayesnode.sample

reachtarunhere · reachtarunhere · commit b86d845e6ed1 · 2016-06-30T02:33:25.000+05:30
diff --git a/tests/test_probability.py b/tests/test_probability.py
@@ -10,13 +10,6 @@ def tests():
     assert cpt.p(True, event) == 0.95
     event = {'Burglary': False, 'Earthquake': True}
     assert cpt.p(False, event) == 0.71
-    # assert BoolCPT({T: 0.2, F: 0.625}).p(False, ['Burglary'], event) == 0.375
-    # assert BoolCPT(0.75).p(False, [], {}) == 0.25
-    # cpt = BoolCPT({True: 0.2, False: 0.7})
-    # assert cpt.rand(['A'], {'A': True}) in [True, False]
-    # cpt = BoolCPT({(True, True): 0.1, (True, False): 0.3,
-    #                (False, True): 0.5, (False, False): 0.7})
-    # assert cpt.rand(['A', 'B'], {'A': True, 'B': False}) in [True, False]
     # #enumeration_ask('Earthquake', {}, burglary)
 
     s = {'A': True, 'B': False, 'C': True, 'D': False}
@@ -87,6 +80,15 @@ def test_enumerate_joint_ask():
 def test_bayesnode_p():
     bn = BayesNode('X', 'Burglary', {T: 0.2, F: 0.625})
     assert bn.p(False, {'Burglary': False, 'Earthquake': True}) == 0.375
+    assert BayesNode('W', '', 0.75).p(False, {'Random': True}) == 0.25
+
+
+def test_bayesnode_sample():
+    X = BayesNode('X', 'Burglary', {T: 0.2, F: 0.625})
+    assert X.sample({'Burglary': False, 'Earthquake': True}) in [True, False]
+    Z = BayesNode('Z', 'P Q', {(True, True): 0.2, (True, False): 0.3,
+                               (False, True): 0.5, (False, False): 0.7})
+    assert Z.sample({'P': True, 'Q': False}) in [True, False]
 
 
 def test_enumeration_ask():
