Skip to content

Commit fb24a54

Browse files
Use bigmemtest.
1 parent 5967c0e commit fb24a54

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

Lib/test/test_random.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)