class DiffieHellmanTest(unittest.TestCase): def setUp(self): # sharedSecret = 17 base = 5 modulo = 23 self.alice = DiffieHellman(modulo, base) self.bob = DiffieHellman(modulo, base) def test(self): aPair = self.alice.partialSessionKeyGen() bPair = self.bob.partialSessionKeyGen() a = self.alice.computeTotalSessionKey(bPair[0]) b = self.bob.computeTotalSessionKey(aPair[0]) self.assertEqual(a,b)
