gh-135326: Test support of __index__ in random.getrandbits() (#135356)

This commit is contained in:
Serhiy Storchaka 2025-06-11 04:51:08 +03:00 committed by GitHub
parent 0f866cbfef
commit 5ae669fc4e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -14,6 +14,15 @@
from fractions import Fraction
from collections import abc, Counter
class MyIndex:
def __init__(self, value):
self.value = value
def __index__(self):
return self.value
class TestBasicOps:
# Superclass with tests common to all generators.
# Subclasses must arrange for self.gen to retrieve the Random instance
@ -809,6 +818,9 @@ def test_getrandbits(self):
self.gen.seed(1234567)
self.assertEqual(self.gen.getrandbits(100),
97904845777343510404718956115)
self.gen.seed(1234567)
self.assertEqual(self.gen.getrandbits(MyIndex(100)),
97904845777343510404718956115)
def test_getrandbits_2G_bits(self):
size = 2**31