1212
1313
1414class DiffieHellmanTest (unittest .TestCase ):
15+
1516 def test_private_key_is_greater_than_1_and_less_than_p (self ):
1617 for prime in [5 , 7 , 11 , 13 , 17 , 19 , 23 , 29 , 31 , 37 , 41 , 43 , 47 ]:
1718 with self .subTest (f"prime={ prime } " ):
1819 key = private_key (prime )
19- self .assertTrue (1 < key < prime , msg = f"{ key } out of range, expected to be >1 and <{ prime } " ) # fmt: skip
20+ self .assertTrue (1 < key < prime , msg = f"{ key } out of range, expected to be >1 and <{ prime } " ) # fmt: skip
2021
2122 def test_private_key_is_random (self ):
2223 """
@@ -30,19 +31,19 @@ def test_can_calculate_public_key_using_private_key(self):
3031 p = 23
3132 g = 5
3233 private_key = 6
33- self .assertEqual (8 , public_key (p , g , private_key , )) # fmt: skip
34+ self .assertEqual (8 , public_key (p , g , private_key , )) # fmt: skip
3435
3536 def test_can_calculate_public_key_when_given_a_different_private_key (self ):
3637 p = 23
3738 g = 5
3839 private_key = 15
39- self .assertEqual (19 , public_key (p , g , private_key , )) # fmt: skip
40+ self .assertEqual (19 , public_key (p , g , private_key , )) # fmt: skip
4041
4142 def test_can_calculate_secret_using_other_party_s_public_key (self ):
4243 p = 23
4344 their_public_key = 19
4445 my_private_key = 6
45- self .assertEqual (2 , secret (p , their_public_key , my_private_key , )) # fmt: skip
46+ self .assertEqual (2 , secret (p , their_public_key , my_private_key , )) # fmt: skip
4647
4748 def test_key_exchange (self ):
4849 p = 23
0 commit comments