def test_intialise_resume(sampler): """Test the initialise method when being used after resuming. In this case the live points are not None """ sampler._flow_proposal = MagicMock() sampler._uninformed_proposal = MagicMock() sampler.populate_live_points = MagicMock() sampler._flow_proposal.initialised = False sampler._uninformed_proposal.initialised = False sampler.live_points = [0.0] sampler.iteration = 100 sampler.maximum_uninformed = 10 sampler.condition = 1.0 sampler.tolerance = 0.1 sampler.initialised = False NestedSampler.initialise(sampler) sampler._flow_proposal.initialise.assert_called_once() sampler._uninformed_proposal.initialise.assert_called_once() sampler.populate_live_points.assert_not_called() assert sampler.initialised is False assert sampler.proposal is sampler._flow_proposal sampler.proposal.configure_pool.assert_called_once()
def test_intialise(sampler): """Test the initialise method when being used without resuming""" sampler._flow_proposal = MagicMock() sampler._uninformed_proposal = MagicMock() sampler.populate_live_points = MagicMock() sampler._flow_proposal.initialised = False sampler._uninformed_proposal.initialised = False sampler.live_points = None sampler.iteration = 0 sampler.maximum_uninformed = 10 sampler.condition = 1.0 sampler.tolerance = 0.1 sampler.initialised = False sampler.uninformed_sampling = True NestedSampler.initialise(sampler) sampler._flow_proposal.initialise.assert_called_once() sampler._uninformed_proposal.initialise.assert_called_once() sampler.populate_live_points.assert_called_once() assert sampler.initialised is True assert sampler.proposal is sampler._uninformed_proposal sampler.proposal.configure_pool.assert_called_once()
def complete_sampler(model, tmpdir): """Complete instance of NestedSampler""" output = tmpdir.mkdir('output') ns = NestedSampler(model, output=output, poolsize=10) ns.initialise() return ns