import unittest

from gPy.Examples import asia, contab
from agm2 import compprob

class Testagm2(unittest.TestCase):

    def test_asia(self):
        val1 = compprob(asia,{'VisitAsia':'no_visit','Smoking':'smoker'})
        val2 = compprob(asia,{'VisitAsia':'no_visit'})
        val3 = compprob(asia,{'Dyspnea':'absent'})
        self.assertAlmostEqual(val1/val2,0.5)
        self.assertAlmostEqual(val1/val3,0.8776138)

    def test_asia2(self):
        from gPy.Models import FR
        fr = FR([cpt*1 for cpt in asia]) 
        val1 = compprob(fr,{'VisitAsia':'no_visit','Smoking':'smoker'})
        val2 = compprob(fr,{'VisitAsia':'no_visit'})
        val3 = compprob(fr,{'Dyspnea':'absent'})
        self.assertAlmostEqual(val1/val2,0.5)
        self.assertAlmostEqual(val1/val3,0.8776138)


    def test_contab(self):
        val1 = compprob(asia,{'Cancer':'absent','Smoking':'smoker'})
        val2 = compprob(asia,{'Bronchitis':'present'})
        val3 = compprob(asia,{'Cancer':'absent','Bronchitis':'present','Smoking':'nonsmoker'})
        self.assertAlmostEqual(val1/val2,1.0)
        self.assertAlmostEqual(val1/val3,3.030303030)



suite = unittest.makeSuite(Testagm2)

if __name__ == '__main__':
    unittest.TextTestRunner(verbosity=2).run(suite)