Skip to content

Commit 4e229f3

Browse files
committed
Update time series tests some more
1 parent 014b601 commit 4e229f3

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

pymc/tests/test_distributions.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -623,7 +623,8 @@ def test_hierarchical_obs_logpt():
623623
assert not any(isinstance(o, RandomVariable) for o in ops)
624624

625625

626-
class TestMatchesScipy:
626+
class LogpChecks():
627+
627628
def check_logp(
628629
self,
629630
pymc_dist,
@@ -775,6 +776,8 @@ def _model_input_dict(model, param_vars, pt):
775776
err_msg=str(invalid_value),
776777
)
777778

779+
class TestMatchesScipy(LogpChecks):
780+
778781
def check_logcdf(
779782
self,
780783
pymc_dist,

pymc/tests/test_distributions_timeseries.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def test_grw_inference(self):
8989
np.testing.assert_allclose([mu, sigma], [recovered_mu, recovered_sigma], atol=0.2)
9090

9191

92-
class TestGRWScipy(td.TestMatchesScipy):
92+
class TestGRWScipy(td.LogpChecks):
9393
# TODO: Test LogP for different inits in its own function
9494

9595
# TODO: Find issue that says GRW wont take vector

0 commit comments

Comments
 (0)