def test_samples(shape, rng, allclose): samples = rng.random_sample(size=shape) d = Samples(samples) dims = None if len(shape) == 1 else shape[1] assert allclose(d.sample(shape[0], dims), samples)
def test_samples_list(samples, allclose): d = Samples(samples) shape = np.array(samples).shape dims = None if len(shape) == 1 else shape[1] assert allclose(d.sample(shape[0], dims), samples)