@@ -811,18 +811,20 @@ def test_getrandbits(self):
811811 97904845777343510404718956115 )
812812
813813 def test_getrandbits_2G_bits (self ):
814+ size = 2 ** 31
814815 self .gen .seed (1234567 )
815- x = self .gen .getrandbits (2 ** 31 )
816- self .assertEqual (x .bit_length (), 2 ** 31 )
817- self .assertEqual (x >> (2 ** 31 - 100 ), 1226514312032729439655761284440 )
816+ x = self .gen .getrandbits (size )
817+ self .assertEqual (x .bit_length (), size )
818818 self .assertEqual (x & (2 ** 100 - 1 ), 890186470919986886340158459475 )
819+ self .assertEqual (x >> (size - 100 ), 1226514312032729439655761284440 )
819820
820- def test_getrandbits_4G_bits (self ):
821+ @support .bigmemtest (size = 2 ** 32 , memuse = 1 / 8 + 2 / 15 , dry_run = False )
822+ def test_getrandbits_4G_bits (self , size ):
821823 self .gen .seed (1234568 )
822- x = self .gen .getrandbits (2 ** 32 )
823- self .assertEqual (x .bit_length (), 2 ** 32 )
824- self .assertEqual (x >> (2 ** 32 - 100 ), 739728759900339699429794460738 )
824+ x = self .gen .getrandbits (size )
825+ self .assertEqual (x .bit_length (), size )
825826 self .assertEqual (x & (2 ** 100 - 1 ), 287241425661104632871036099814 )
827+ self .assertEqual (x >> (size - 100 ), 739728759900339699429794460738 )
826828
827829 def test_randrange_uses_getrandbits (self ):
828830 # Verify use of getrandbits by randrange
@@ -980,10 +982,11 @@ def test_randbytes_getrandbits(self):
980982 self .assertEqual (self .gen .randbytes (n ),
981983 gen2 .getrandbits (n * 8 ).to_bytes (n , 'little' ))
982984
983- def test_randbytes_256M (self ):
985+ @support .bigmemtest (size = 2 ** 29 , memuse = 1 + 16 / 15 , dry_run = False )
986+ def test_randbytes_256M (self , size ):
984987 self .gen .seed (2849427419 )
985- x = self .gen .randbytes (2 ** 29 )
986- self .assertEqual (len (x ), 2 ** 29 )
988+ x = self .gen .randbytes (size )
989+ self .assertEqual (len (x ), size )
987990 self .assertEqual (x [:12 ].hex (), 'f6fd9ae63855ab91ea238b4f' )
988991 self .assertEqual (x [- 12 :].hex (), '0e7af69a84ee99bf4a11becc' )
989992
0 commit comments