diff --git a/test/core/sampler/ptemcee_test.py b/test/core/sampler/ptemcee_test.py index 70abcb0af21e3a9a55ca486a1c0c0981a66dc2f7..08439c814878878583d191b2170a5ac36a5e751b 100644 --- a/test/core/sampler/ptemcee_test.py +++ b/test/core/sampler/ptemcee_test.py @@ -28,36 +28,36 @@ class TestPTEmcee(unittest.TestCase): def test_default_kwargs(self): expected = dict( - ntemps=20, - nwalkers=200, + ntemps=10, + nwalkers=100, Tmax=None, betas=None, a=2.0, adaptation_lag=10000, adaptation_time=100, random=None, - adapt=True, + adapt=False, swap_ratios=False, ) self.assertDictEqual(expected, self.sampler.kwargs) def test_translate_kwargs(self): expected = dict( - ntemps=20, - nwalkers=200, + ntemps=10, + nwalkers=100, Tmax=None, betas=None, a=2.0, adaptation_lag=10000, adaptation_time=100, random=None, - adapt=True, + adapt=False, swap_ratios=False, ) for equiv in bilby.core.sampler.base_sampler.MCMCSampler.nwalkers_equiv_kwargs: new_kwargs = self.sampler.kwargs.copy() del new_kwargs["nwalkers"] - new_kwargs[equiv] = 200 + new_kwargs[equiv] = 100 self.sampler.kwargs = new_kwargs self.assertDictEqual(expected, self.sampler.kwargs)