def test_insertion_indices_p_none(mock_fn, sampler): """Test computing the distribution of insertion indices if p is None""" sampler.rolling_p = [] sampler.insertion_indices = \ np.random.randint(sampler.nlive, size=2 * sampler.nlive) NestedSampler.check_insertion_indices(sampler, rolling=True) assert len(sampler.rolling_p) == 0
def test_insertion_indices_save(mock_fn, mock_save, filename, sampler): """Test saving the insertion indices""" sampler.output = './' sampler.insertion_indices = \ np.random.randint(sampler.nlive, size=2 * sampler.nlive) NestedSampler.check_insertion_indices(sampler, rolling=False, filename=filename) if filename: mock_save.assert_called_once_with('./file.txt', sampler.insertion_indices, newline='\n', delimiter=' ')
def test_insertion_indices(mock_fn, rolling, sampler): """Test computing the distribution of insertion indices""" sampler.rolling_p = [] sampler.insertion_indices = \ np.random.randint(sampler.nlive, size=2 * sampler.nlive) NestedSampler.check_insertion_indices(sampler, rolling=rolling) if rolling: assert len(sampler.rolling_p) == 1 np.testing.assert_array_equal( mock_fn.call_args_list[0][0][0], sampler.insertion_indices[-sampler.nlive:]) else: mock_fn.assert_called_once_with(sampler.insertion_indices, sampler.nlive)