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']