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_2(): config = sampler_config_2() sampler = SamplerManager(config) assert sampler.is_empty() is False assert isinstance(sampler.samplers["angle"], GaussianSampler) # Check angle gaussian sampler assert sampler.samplers["angle"].mean == config["angle"]["mean"] assert sampler.samplers["angle"].st_dev == config["angle"]["st_dev"]
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)