diff --git a/tests/test_mcmc.py b/tests/test_mcmc.py index d50a51c..b980ff0 100644 --- a/tests/test_mcmc.py +++ b/tests/test_mcmc.py @@ -64,10 +64,12 @@ def fix_state(request): @pytest.fixture( params=[(0, 4000, 1), (2000, 4000, 5), (0, 6000, 10), (2000, 6000, 1)], - ids=["n_burn=0,n_iter=4000, n_thin=1", - "n_burn=non-zero,n_iter=4000,n_thin=5", - "n_burn=0,n_iter=6000, n_thin=10", - "n_burn=non-zero,n_iter=6000,n_thin=1"], + ids=[ + "n_burn=0,n_iter=4000, n_thin=1", + "n_burn=non-zero,n_iter=4000,n_thin=5", + "n_burn=0,n_iter=6000, n_thin=10", + "n_burn=non-zero,n_iter=6000,n_thin=1", + ], name="mcmc_settings", ) def fix_mcmc_settings(request): @@ -108,10 +110,14 @@ def mock_log_p(self, current_state): monkeypatch.setattr(NormalGamma, "store", mock_store) monkeypatch.setattr(Model, "log_p", mock_log_p) - M = MCMC(state, sampler, model, - n_burn=mcmc_settings["nburn"], - n_iter=mcmc_settings["niter"], - n_thin=mcmc_settings["nthin"]) + M = MCMC( + state, + sampler, + model, + n_burn=mcmc_settings["nburn"], + n_iter=mcmc_settings["niter"], + n_thin=mcmc_settings["nthin"], + ) M.store["count"] = 0 M.run_mcmc() assert M.state["count"] == (M.n_iter + M.n_burn) * len(sampler) * M.n_thin