def test_n_samples(): sampler1 = BasicSweep({'a': [1, 2, 3], 'b': [4, 5, 6]}) sampler2 = BasicSweep({'a': [1, 2, 3], 'b': [4, 5, 6]}) multi = MultiSampler(sampler1, sampler2) assert(multi.n_samples() == 81)
def test_init(): sweep = BasicSweep({'a': [1, 2, 3], 'b': [4, 5, 6]}, 10) with pytest.raises(StopIteration): sweep.__next__()
def test_basic_sweep_single_list(): assert (len(list(BasicSweep({'a': [1, 2, 3]}))) == 3)
def basic_sweep_sampler(): return BasicSweep({'a': [1, 2, 3], 'b': [4, 5, 6]})