def testPeekWithWrongChoice(self): doors['chosen'] = 2 openDoor() self.assertEquals(doors['open'], 3) doors['chosen'] = 3 openDoor() self.assertEquals(doors['open'], 2)
def testRandomPeek(self): choose() openDoor() self.assertNotEquals(doors['open'], doors['chosen']) self.assertNotEquals(doors['open'], doors['prize'])
def testPeekWithRightChoice(self): doors['chosen'] = 1 openDoor() self.assertIn(doors['open'], [2,3])