diff --git a/pymc/tests/test_model.py b/pymc/tests/test_model.py index fc03bf2182..edb2888834 100644 --- a/pymc/tests/test_model.py +++ b/pymc/tests/test_model.py @@ -1138,11 +1138,11 @@ def setup_method(self): _, self.model, _ = simple_model() def test_profile_model(self): - assert self.model.profile(self.model.logp()).fct_call_time > 0 + assert self.model.profile(self.model.logp(), n=3000).fct_call_time > 0 def test_profile_variable(self): rv = self.model.basic_RVs[0] - assert self.model.profile(self.model.logp(vars=[rv], sum=False)).fct_call_time + assert self.model.profile(self.model.logp(vars=[rv], sum=False), n=3000).fct_call_time > 0 def test_profile_count(self): count = 1005