class TestMarkIV(unittest.TestCase):
def setUp(self):
self.mark2 = MarkII(4, 7)
self.mark4 = MarkIV(4, 7)
def testInitialEncode(self):
self.assertEquals(self.mark4.encode('H'),
self.mark2.encode('H'),
"Initial encoding of MarkIV should be the same as the MarkII")
def testSecondEncode(self):
self.assertNotEquals(self.mark4.encode('HE'),
self.mark2.encode('HE'),
"Second encoding of MarkIV should be different from the MarkII")
def testQuestionThree(self):
self.assertEquals(self.mark4.encode('The white cliffs of Alghero are visible at night'),
'JMl0kBp?20QixoivSc.2"vvmls8KOk"0jA,4kgt0OmUb,pm.')
def testDecodeString(self):
self.assertEquals(self.mark4.decode('JMl0kBp?20QixoivSc.2"vvmls8KOk"0jA,4kgt0OmUb,pm.'),
'The white cliffs of Alghero are visible at night')
def setUp(self):
self.mark2 = MarkII(4, 7)
self.mark4 = MarkIV(4, 7)
