Skip to content

Commit 7e8e838

Browse files
committed
Revert reduced test n_samples due to speed issues
1 parent fe67c98 commit 7e8e838

File tree

1 file changed

+6
-16
lines changed

1 file changed

+6
-16
lines changed

pymc3/tests/test_distributions.py

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1137,13 +1137,17 @@ def test_beta(self):
11371137
{"alpha": Rplus, "beta": Rplus},
11381138
lambda value, alpha, beta: sp.beta.logpdf(value, alpha, beta),
11391139
)
1140-
self.check_logp(Beta, Unit, {"mu": Unit, "sigma": Rplus}, beta_mu_sigma)
1140+
self.check_logp(
1141+
Beta,
1142+
Unit,
1143+
{"mu": Unit, "sigma": Rplus},
1144+
beta_mu_sigma,
1145+
)
11411146
self.check_logcdf(
11421147
Beta,
11431148
Unit,
11441149
{"alpha": Rplus, "beta": Rplus},
11451150
lambda value, alpha, beta: sp.beta.logcdf(value, alpha, beta),
1146-
n_samples=10,
11471151
decimal=select_by_precision(float64=5, float32=3),
11481152
)
11491153

@@ -1266,20 +1270,17 @@ def scipy_mu_alpha_logcdf(value, mu, alpha):
12661270
Nat,
12671271
{"mu": Rplus, "alpha": Rplus},
12681272
scipy_mu_alpha_logcdf,
1269-
n_samples=5,
12701273
)
12711274
self.check_logcdf(
12721275
NegativeBinomial,
12731276
Nat,
12741277
{"p": Unit, "n": Rplus},
12751278
lambda value, p, n: sp.nbinom.logcdf(value, n, p),
1276-
n_samples=5,
12771279
)
12781280
self.check_selfconsistency_discrete_logcdf(
12791281
NegativeBinomial,
12801282
Nat,
12811283
{"mu": Rplus, "alpha": Rplus},
1282-
n_samples=10,
12831284
)
12841285

12851286
@pytest.mark.xfail(reason="Distribution not refactored yet")
@@ -1338,7 +1339,6 @@ def test_lognormal(self):
13381339
Rplus,
13391340
{"mu": R, "sigma": Rplusbig},
13401341
lambda value, mu, sigma: floatX(sp.lognorm.logpdf(value, sigma, 0, np.exp(mu))),
1341-
n_samples=5, # Just testing alternative parametrization
13421342
)
13431343
self.check_logcdf(
13441344
Lognormal,
@@ -1351,7 +1351,6 @@ def test_lognormal(self):
13511351
Rplus,
13521352
{"mu": R, "sigma": Rplusbig},
13531353
lambda value, mu, sigma: sp.lognorm.logcdf(value, sigma, 0, np.exp(mu)),
1354-
n_samples=5, # Just testing alternative parametrization
13551354
)
13561355

13571356
def test_t(self):
@@ -1366,21 +1365,18 @@ def test_t(self):
13661365
R,
13671366
{"nu": Rplus, "mu": R, "sigma": Rplus},
13681367
lambda value, nu, mu, sigma: sp.t.logpdf(value, nu, mu, sigma),
1369-
n_samples=5, # Just testing alternative parametrization
13701368
)
13711369
self.check_logcdf(
13721370
StudentT,
13731371
R,
13741372
{"nu": Rplus, "mu": R, "lam": Rplus},
13751373
lambda value, nu, mu, lam: sp.t.logcdf(value, nu, mu, lam ** -0.5),
1376-
n_samples=10, # relies on slow incomplete beta
13771374
)
13781375
self.check_logcdf(
13791376
StudentT,
13801377
R,
13811378
{"nu": Rplus, "mu": R, "sigma": Rplus},
13821379
lambda value, nu, mu, sigma: sp.t.logcdf(value, nu, mu, sigma),
1383-
n_samples=5, # Just testing alternative parametrization
13841380
)
13851381

13861382
def test_cauchy(self):
@@ -1557,13 +1553,11 @@ def test_binomial(self):
15571553
Nat,
15581554
{"n": NatSmall, "p": Unit},
15591555
lambda value, n, p: sp.binom.logcdf(value, n, p),
1560-
n_samples=10,
15611556
)
15621557
self.check_selfconsistency_discrete_logcdf(
15631558
Binomial,
15641559
Nat,
15651560
{"n": NatSmall, "p": Unit},
1566-
n_samples=10,
15671561
)
15681562

15691563
@pytest.mark.xfail(reason="checkd tests has not been refactored")
@@ -1766,14 +1760,12 @@ def logcdf_fn(value, psi, mu, alpha):
17661760
Nat,
17671761
{"psi": Unit, "mu": Rplusbig, "alpha": Rplusbig},
17681762
logcdf_fn,
1769-
n_samples=10,
17701763
)
17711764

17721765
self.check_selfconsistency_discrete_logcdf(
17731766
ZeroInflatedNegativeBinomial,
17741767
Nat,
17751768
{"psi": Unit, "mu": Rplusbig, "alpha": Rplusbig},
1776-
n_samples=10,
17771769
)
17781770

17791771
@pytest.mark.xfail(reason="Test not refactored yet")
@@ -1806,14 +1798,12 @@ def logcdf_fn(value, psi, n, p):
18061798
Nat,
18071799
{"psi": Unit, "n": NatSmall, "p": Unit},
18081800
logcdf_fn,
1809-
n_samples=10,
18101801
)
18111802

18121803
self.check_selfconsistency_discrete_logcdf(
18131804
ZeroInflatedBinomial,
18141805
Nat,
18151806
{"n": NatSmall, "p": Unit, "psi": Unit},
1816-
n_samples=10,
18171807
)
18181808

18191809
@pytest.mark.parametrize("n", [1, 2, 3])

0 commit comments

Comments
 (0)