summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_random.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_random.py')
-rw-r--r--Lib/test/test_random.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/Lib/test/test_random.py b/Lib/test/test_random.py
index 32e7868..f32d592 100644
--- a/Lib/test/test_random.py
+++ b/Lib/test/test_random.py
@@ -111,6 +111,21 @@ class TestBasicOps:
self.assertEqual(choice([50]), 50)
self.assertIn(choice([25, 75]), [25, 75])
+ def test_choice_with_numpy(self):
+ # Accommodation for NumPy arrays which have disabled __bool__().
+ # See: https://github.com/python/cpython/issues/100805
+ choice = self.gen.choice
+
+ class NA(list):
+ "Simulate numpy.array() behavior"
+ def __bool__(self):
+ raise RuntimeError
+
+ with self.assertRaises(IndexError):
+ choice(NA([]))
+ self.assertEqual(choice(NA([50])), 50)
+ self.assertIn(choice(NA([25, 75])), [25, 75])
+
def test_sample(self):
# For the entire allowable range of 0 <= k <= N, validate that
# the sample is of the correct length and contains only unique items