def test_sampling(self):
"""Make sure Fidelity simply returns `high`"""
dim = Fidelity("epoch", 1, 2)
assert dim.sample() == [2]
dim = Fidelity("epoch", 1, 5)
assert dim.sample() == [5]
dim = Fidelity("epoch", 1, 5)
assert dim.sample(4) == [5] * 4
def test_sampling(self):
"""Make sure Fidelity simply returns `fidelity`"""
dim = Fidelity('epoch')
assert dim.sample() == ['fidelity']
