def test_add_plot_posterior_option_group():
parser = argparse.ArgumentParser()
option_utils.add_plot_posterior_option_group(parser)
args = parser.parse_args([
'--plot-marginal',
'--marginal-percentiles',
'5',
'95',
'--plot-scatter',
'--plot-density',
'--plot-contours',
'--contour-percentiles',
'5',
'95',
'--mins',
'a:1',
'b:2',
'--maxs',
'a:2',
'b:3',
'--expected-parameters',
'a:1',
'b:2',
'--expected-parameters-color',
'b',
'--plot-injection-parameters',
'--injection-hdf-group',
'blah',
])
for arg in ('marginal_percentiles', 'contour_percentiles'):
assert list(map(type, getattr(args, arg))) == [float, float]
def test_expected_parameters_from_cli():
parser = argparse.ArgumentParser()
option_utils.add_plot_posterior_option_group(parser)
args = parser.parse_args([
'--expected-parameters',
'a:1',
'b:2',
])
assert option_utils.expected_parameters_from_cli(args) == {'a': 1, 'b': 2}
def test_plot_ranges_from_cli():
parser = argparse.ArgumentParser()
option_utils.add_plot_posterior_option_group(parser)
args = parser.parse_args([
'--mins',
'a:1',
'b:2',
'--maxs',
'a:2',
'b:3',
])
assert option_utils.plot_ranges_from_cli(args) == ({
'a': 1,
'b': 2
}, {
'a': 2,
'b': 3
})