From: Serhiy Storchaka <storchaka@gmail.com>
Date: Wed, 11 Jun 2025 01:51:08 +0000 (+0300)
Subject: gh-135326: Test support of __index__ in random.getrandbits() (#135356)
X-Git-Tag: v3.15.0a1~1331
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=5ae669fc4e674968529cc32f7f31d14dddd76607;p=thirdparty%2FPython%2Fcpython.git

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

diff --git a/Lib/test/test_random.py b/Lib/test/test_random.py
index 54910cd8054a..31ebcb3b8b0b 100644
--- a/Lib/test/test_random.py
+++ b/Lib/test/test_random.py
@@ -14,6 +14,15 @@ from test import support
 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 @@ class MersenneTwister_TestBasicOps(TestBasicOps, unittest.TestCase):
         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