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])
