def test_update_raises(self):
settings = fps.CosmicFish_PlotSettings()
assert_raises( ValueError, settings.update , 'test' )
# ***************************************************************************************
def test_init(self):
set = fps.CosmicFish_PlotSettings(do_legend=False)
test_plotter_1 = fp.CosmicFishPlotter(fishers=self.fisher_list_test,
settings=set)
assert test_plotter_1.plot_settings.do_legend == False
test_plotter_2 = fp.CosmicFishPlotter(fishers=self.fisher_list_test,
settings={'do_legend': False})
assert test_plotter_2.plot_settings.do_legend == False
fisher_list = fishers.get_fisher_matrix()
for fish in fisher_list[1:]:
fisher_list[0] = fisher_list[0] + fish
fisher_list[0].name = args.sum_fish
fishers_temp.add_fisher_matrix(fisher_list[0])
fishers = fishers_temp
if args.eliminate:
if params is None:
raise ValueError(
'Avoiding marginalization only works if parameters are specified'
)
fishers = fishers.reshuffle(params=params)
# set up the plot settings
plot_settings = fps.CosmicFish_PlotSettings()
# set up the plotter
plotter = fp.CosmicFishPlotter(settings=plot_settings, fishers=fishers)
if params is not None:
params = [list(i) for i in it.combinations(params, 2)]
if len(params) == 0:
raise ValueError('Not enough parameters for 2D plot.')
plotter.new_plot()
plotter.plot2D(params=params)
plotter.export(outroot + '.' + output_format)
plotter.close_plot()
def test_update_kwargs(self):
settings = fps.CosmicFish_PlotSettings()
settings.update( do_legend=False, D2_show_xaxis_label=False )
assert settings.do_legend==False
assert settings.D2_show_xaxis_label == False
def test_update_dict(self):
settings = fps.CosmicFish_PlotSettings()
settings.update({'do_legend':False})
assert settings.do_legend==False
def test_init_settings_with_kwargs(self):
settings = fps.CosmicFish_PlotSettings( do_legend=False, D2_show_xaxis_label=False )
assert settings.do_legend==False
assert settings.D2_show_xaxis_label == False
def test_init_settings_with_dict(self):
settings = fps.CosmicFish_PlotSettings({'do_legend':False})
assert settings.do_legend==False
def test_init_settings_default(self):
settings = fps.CosmicFish_PlotSettings()