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)
