Add larger shapes to some tests
@@ -190,6 +190,7 @@ standard_normal_params = [
{'size': None},
{'size': (1, 2, 3)},
{'size': 3},
+ {'size': (1000, 1000)},
{'size': (3, 3)},
{'size': ()}]
@@ -241,6 +242,9 @@ class Poisson:
self.check_ks(0.05)(
lam=self.lam, size=2000)
+ def test_poisson_large(self):
+ self.generate(lam=self.lam, size=(1000, 1000))
+
gamma_params = [
{'shape': 0.5, 'scale': 0.5},
