def test_inverse(self): f_my_private = top.f_private(self.privkey) f_my_public = top.f_public(self.privkey) self.assertEqual(MSG1, f_my_public(f_my_private(MSG1))) self.assertEqual(MSG1, f_my_private(f_my_public(MSG1))) self.assertNotEqual(MSG1, f_my_public(f_my_public(MSG1))) self.assertNotEqual(MSG1, f_my_private(f_my_private(MSG1)))
def test_private(self): f_my_private = top.f_private(self.privkey) self.assertTrue(f_my_private(MSG1)) f_my_private2 = top.f_private(self.pubkey) # cannot use private function on a public key self.assertRaises(Exception, f_my_private2, MSG1)