Ejemplos de NestedSampler.populate_live_points en Python

Ejemplo n.º 1

Archivo: test_live_points.py Proyecto: mj-will/nessai

def test_populate_live_points_nans(sampler):
    """Test popluting the live points with NaN values"""
    new_points = sampler.model.new_point(sampler.nlive + 1)
    new_points['logL'][4] = np.nan
    sampler.yield_sample = MagicMock(
        return_value=iter(zip(np.ones(sampler.nlive + 1), new_points)))
    NestedSampler.populate_live_points(sampler)
    assert len(sampler.live_points) == sampler.nlive
    assert not np.isnan(sampler.live_points['logL']).any()

Ejemplo n.º 2

Archivo: test_live_points.py Proyecto: mj-will/nessai

def test_populate_live_points(sampler):
    """Test popluting the live points"""
    sampler.yield_sample = MagicMock(return_value=iter(
        zip(np.ones(sampler.nlive), sampler.model.new_point(sampler.nlive))))
    NestedSampler.populate_live_points(sampler)
    assert len(sampler.live_points) == sampler.nlive