def test_convert_to_samples(proposal):
"""Test convert to sample without the prime prior"""
samples = numpy_array_to_live_points(np.random.randn(10, 4), ['x', 'y'])
proposal.use_x_prime_prior = False
proposal.model = MagicMock()
proposal.model.names = ['x']
proposal.model.log_prior = MagicMock(return_value=np.ones(10))
out_samples = FlowProposal.convert_to_samples(proposal, samples, plot=True)
assert out_samples.dtype.names == ('x', 'logP', 'logL')
def test_convert_to_samples_with_prime(mock_plot, proposal):
"""Test convert to sample with the prime prior"""
samples = numpy_array_to_live_points(np.random.randn(10, 4), ['x', 'y'])
proposal.use_x_prime_prior = True
proposal.model = MagicMock()
proposal.model.names = ['x']
proposal.model.log_prior = MagicMock(return_value=np.ones(10))
proposal._plot_pool = True
proposal.training_data_prime = 'data'
proposal.output = './'
proposal.populated_count = 1
proposal.inverse_rescale = MagicMock(return_value=(samples, None))
out_samples = FlowProposal.convert_to_samples(proposal, samples, plot=True)
mock_plot.assert_called_once_with('data',
samples,
labels=['live points', 'pool'],
filename='.//pool_prime_1.png')
proposal.inverse_rescale.assert_called_once()
assert out_samples.dtype.names == ('x', 'logP', 'logL')