#format python

import unittest
import YootNori

class MockRandom:
    def random(self):
        return 0.5

YootNori.random.random = MockRandom().random

class TestYootNori(unittest.TestCase):
    def testOneYoot(self):
        theGame = YootNori.DoingYootNori(0.6)
        self.assertEquals(1, theGame.throwOneYoot())
    def testFourYoot(self):
        theGame = YootNori.DoingYootNori(0.6)
        self.assertEquals([1,1,1,1], theGame.throwFourYoot())
    def testJudgeYoot(self):
        theGame = YootNori.DoingYootNori(0.6)
        self.assertEquals('MO', theGame.judgeYoot([1,1,1,1]))
    def testThrow(self):
        theGame = YootNori.DoingYootNori(0.6)
        self.assertEquals('MO', theGame.throw())

class TestStatYootNori(unittest.TestCase):
    def testTotalProbability(self):
        theGame = YootNori.DoingYootNori(0.5)
        totalProb = 0
        for aYoot in YootNori.mapOfYootName.keys():
            totalProb += theGame.getProbability(aYoot)
            #print aYoot, theGame.getProbability(aYoot)
        self.assertEquals(1, totalProb)

if __name__ == '__main__':
    unittest.main(argv=('', '-v'))