def test_name_plot_options_simple(self):
"""
Check set_plot_options can modify given attribute
"""
data_list = setup_McStasData_array()
name_plot_options("Hero", data_list, colormap="Oranges")
hero_object = name_search("Hero", data_list)
self.assertEqual(hero_object.plot_options.colormap, "Oranges")
def test_name_plot_options_simple(self):
"""
Test simple case
"""
data_list = setup_McStasData_array()
name_plot_options("Hero", data_list, colormap="very hot")
hero_object = name_search("Hero", data_list)
self.assertEqual(hero_object.plot_options.colormap, "very hot")
def test_name_plot_options_duplicate(self):
"""
Test case where several datasets are modified
"""
data_list = setup_McStasData_array()
hero_object = set_dummy_McStasData_2d("Hero")
hero_object.metadata.dimension = 321
hero_object.plot_options.colormap = "absurdly hot"
data_list.append(hero_object)
name_plot_options("Hero", data_list, colormap="cold")
results = name_search("Hero", data_list)
self.assertEqual(len(results), 2)
self.assertEqual(results[0].plot_options.colormap, "cold")
self.assertEqual(results[1].plot_options.colormap, "cold")
def test_name_plot_options_duplicate(self):
"""
Test case where several McStasData objects are modified since
the internal name_search finds multiple matches
"""
data_list = setup_McStasData_array()
hero_object = set_dummy_McStasDataBinned_2d("Hero")
hero_object.metadata.dimension = 321
hero_object.plot_options.colormap = "absurdly hot"
data_list.append(hero_object)
name_plot_options("Hero", data_list, colormap="Blues")
results = name_search("Hero", data_list)
self.assertEqual(len(results), 2)
self.assertEqual(results[0].plot_options.colormap, "Blues")
self.assertEqual(results[1].plot_options.colormap, "Blues")