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")