MAINT: improve rvs of randint in scipy.stats

scipy/stats/_discrete_distns.py

@@ -741,6 +741,9 @@ def _stats(self, low, high):
 
     def _rvs(self, low, high):
         """An array of *size* random integers >= ``low`` and < ``high``."""
+        if np.asarray(low).size == 1 and np.asarray(high).size == 1:
+            # no need to vectorize in that case
+            return self._random_state.randint(low, high, size=self._size)
         if self._size is not None:
             # NumPy's RandomState.randint() doesn't broadcast its arguments.
             # Use `broadcast_to()` to extend the shapes of low and high

scipy/stats/tests/test_distributions.py

@@ -194,7 +194,7 @@ def test_pdf(self):
         assert_array_almost_equal(vals, out)
 
     def test_cdf(self):
-        x = np.linspace(0,36,100)
+        x = np.linspace(0, 36, 100)
         k = numpy.floor(x)
         out = numpy.select([k >= 30, k >= 5], [1.0, (k-5.0+1)/(30-5.0)], 0)
         vals = stats.randint.cdf(x, 5, 30)