def test_empty_samplers(): empty_sampler = SamplerManager({}) assert empty_sampler.is_empty() empty_cur_sample = empty_sampler.sample_all() assert empty_cur_sample == {} none_sampler = SamplerManager(None) assert none_sampler.is_empty() none_cur_sample = none_sampler.sample_all() assert none_cur_sample == {}
def test_sampler_config_1(): config = sampler_config_1() sampler = SamplerManager(config) assert sampler.is_empty() is False assert isinstance(sampler.samplers["mass"], UniformSampler) assert isinstance(sampler.samplers["gravity"], MultiRangeUniformSampler) cur_sample = sampler.sample_all() # Check uniform sampler for mass assert sampler.samplers["mass"].min_value == config["mass"]["min_value"] assert sampler.samplers["mass"].max_value == config["mass"]["max_value"] assert config["mass"]["min_value"] <= cur_sample["mass"] assert config["mass"]["max_value"] >= cur_sample["mass"] # Check multirange_uniform sampler for gravity assert sampler.samplers["gravity"].intervals == config["gravity"][ "intervals"] assert check_value_in_intervals(cur_sample["gravity"], sampler.samplers["gravity"].intervals)