def setUp(self): self.app = QtGui.QApplication(sys.argv) self.image_model = ImgModel() self.calibration_model = CalibrationModel() self.calibration_model.load(os.path.join(data_path, 'LaB6_40keV_MarCCD.poni')) self.spectrum_model = SpectrumModel() self.phase_model = PhaseModel() self.widget = IntegrationWidget() self.widget.spectrum_view._auto_range = True self.phase_tw = self.widget.phase_tw self.spectrum_controller = SpectrumController({}, self.widget, self.image_model, None, self.calibration_model, self.spectrum_model) self.controller = PhaseController({}, self.widget, self.calibration_model, self.spectrum_model, self.phase_model) self.spectrum_model.load_spectrum(os.path.join(data_path, 'spectrum_001.xy'))
def setUp(self): self.x = np.linspace(0.1, 15, 100) self.y = np.sin(self.x) self.spectrum = Spectrum(self.x, self.y) self.spectrum_model = SpectrumModel()
class PhaseControllerTest(unittest.TestCase): def setUp(self): self.app = QtGui.QApplication(sys.argv) self.image_model = ImgModel() self.calibration_model = CalibrationModel() self.calibration_model.load(os.path.join(data_path, 'LaB6_40keV_MarCCD.poni')) self.spectrum_model = SpectrumModel() self.phase_model = PhaseModel() self.widget = IntegrationWidget() self.widget.spectrum_view._auto_range = True self.phase_tw = self.widget.phase_tw self.spectrum_controller = SpectrumController({}, self.widget, self.image_model, None, self.calibration_model, self.spectrum_model) self.controller = PhaseController({}, self.widget, self.calibration_model, self.spectrum_model, self.phase_model) self.spectrum_model.load_spectrum(os.path.join(data_path, 'spectrum_001.xy')) def tearDown(self): del self.app def test_manual_deleting_phases(self): self.load_phases() QtGui.QApplication.processEvents() self.assertEqual(self.phase_tw.rowCount(), 6) self.assertEqual(len(self.phase_model.phases), 6) self.assertEqual(len(self.widget.spectrum_view.phases), 6) self.assertEqual(self.phase_tw.currentRow(), 5) self.controller.remove_btn_click_callback() self.assertEqual(self.phase_tw.rowCount(), 5) self.assertEqual(len(self.phase_model.phases), 5) self.assertEqual(len(self.widget.spectrum_view.phases), 5) self.assertEqual(self.phase_tw.currentRow(), 4) self.widget.select_phase(1) self.controller.remove_btn_click_callback() self.assertEqual(self.phase_tw.rowCount(), 4) self.assertEqual(len(self.phase_model.phases), 4) self.assertEqual(len(self.widget.spectrum_view.phases), 4) self.assertEqual(self.phase_tw.currentRow(), 1) self.widget.select_phase(0) self.controller.remove_btn_click_callback() self.assertEqual(self.phase_tw.rowCount(), 3) self.assertEqual(len(self.phase_model.phases), 3) self.assertEqual(len(self.widget.spectrum_view.phases), 3) self.assertEqual(self.phase_tw.currentRow(), 0) self.controller.remove_btn_click_callback() self.controller.remove_btn_click_callback() self.controller.remove_btn_click_callback() self.assertEqual(self.phase_tw.rowCount(), 0) self.assertEqual(len(self.phase_model.phases), 0) self.assertEqual(len(self.widget.spectrum_view.phases), 0) self.assertEqual(self.phase_tw.currentRow(), -1) self.controller.remove_btn_click_callback() self.assertEqual(self.phase_tw.rowCount(), 0) self.assertEqual(len(self.phase_model.phases), 0) self.assertEqual(len(self.widget.spectrum_view.phases), 0) self.assertEqual(self.phase_tw.currentRow(), -1) def test_automatic_deleting_phases(self): self.load_phases() self.load_phases() self.assertEqual(self.phase_tw.rowCount(), 12) self.assertEqual(len(self.phase_model.phases), 12) self.assertEqual(len(self.widget.spectrum_view.phases), 12) self.controller.clear_phases() self.assertEqual(self.phase_tw.rowCount(), 0) self.assertEqual(len(self.phase_model.phases), 0) self.assertEqual(len(self.widget.spectrum_view.phases), 0) self.assertEqual(self.phase_tw.currentRow(), -1) multiplier = 1 for dummy_index in range(multiplier): self.load_phases() self.assertEqual(self.phase_tw.rowCount(), multiplier * 6) self.controller.clear_phases() self.assertEqual(self.phase_tw.rowCount(), 0) self.assertEqual(len(self.phase_model.phases), 0) self.assertEqual(len(self.widget.spectrum_view.phases), 0) self.assertEqual(self.phase_tw.currentRow(), -1) def test_pressure_change(self): self.load_phases() pressure = 200 self.widget.phase_pressure_sb.setValue(200) for ind, phase in enumerate(self.phase_model.phases): self.assertEqual(phase.pressure, pressure) self.assertEqual(self.widget.get_phase_pressure(ind), pressure) def test_temperature_change(self): self.load_phases() temperature = 1500 self.widget.phase_temperature_sb.setValue(temperature) for ind, phase in enumerate(self.phase_model.phases): if phase.has_thermal_expansion(): self.assertEqual(phase.temperature, temperature) self.assertEqual(self.widget.get_phase_temperature(ind), temperature) else: self.assertEqual(phase.temperature, 298) self.assertEqual(self.widget.get_phase_temperature(ind), None) def test_apply_to_all_for_new_added_phase_in_table_widget(self): temperature = 1500 pressure = 200 self.widget.phase_temperature_sb.setValue(temperature) self.widget.phase_pressure_sb.setValue(pressure) self.load_phases() for ind, phase in enumerate(self.phase_model.phases): self.assertEqual(phase.pressure, pressure) self.assertEqual(self.widget.get_phase_pressure(ind), pressure) if phase.has_thermal_expansion(): self.assertEqual(phase.temperature, temperature) self.assertEqual(self.widget.get_phase_temperature(ind), temperature) else: self.assertEqual(phase.temperature, 298) self.assertEqual(self.widget.get_phase_temperature(ind), None) def test_apply_to_all_for_new_added_phase_d_positions(self): pressure = 50 self.load_phase('au_Anderson.jcpds') self.widget.phase_pressure_sb.setValue(pressure) self.load_phase('au_Anderson.jcpds') reflections1 = self.phase_model.get_lines_d(0) reflections2 = self.phase_model.get_lines_d(1) self.assertTrue(np.array_equal(reflections1, reflections2)) def test_to_not_show_lines_in_legend(self): self.load_phases() self.phase_tw.selectRow(1) QTest.mouseClick(self.widget.phase_del_btn, QtCore.Qt.LeftButton) self.widget.spectrum_view.hide_phase(1) def test_auto_scaling_of_lines_in_spectrum_view(self): spectrum_view = self.widget.spectrum_view spectrum_view_range = spectrum_view.view_box.viewRange() spectrum_y = spectrum_view.plot_item.getData()[1] expected_maximum_height = np.max(spectrum_y) - spectrum_view_range[1][0] self.load_phase('au_Anderson.jcpds') phase_plot = spectrum_view.phases[0] line_heights = [] for line in phase_plot.line_items: line_data = line.getData() height = line_data[1][1]-line_data[1][0] line_heights.append(height) self.assertAlmostEqual(expected_maximum_height, np.max(line_heights)) spectrum_view_range = spectrum_view.view_box.viewRange() spectrum_y = spectrum_view.plot_item.getData()[1] expected_maximum_height = np.max(spectrum_y) - spectrum_view_range[1][0] self.assertAlmostEqual(expected_maximum_height, np.max(line_heights)) def test_line_height_in_spectrum_view_after_zooming(self): spectrum_view = self.widget.spectrum_view self.load_phase('au_Anderson.jcpds') spectrum_view.view_box.setRange(xRange=[17,30]) spectrum_view.emit_sig_range_changed() phase_plot = spectrum_view.phases[0] line_heights = [] for line in phase_plot.line_items: line_data = line.getData() if (line_data[0][0] > 17) and (line_data[0][1]<30): height = line_data[1][1]-line_data[1][0] line_heights.append(height) spectrum_view_range = spectrum_view.view_box.viewRange() spectrum_x, spectrum_y = spectrum_view.plot_item.getData() spectrum_y_max_in_range = np.max(spectrum_y[(spectrum_x > spectrum_view_range[0][0]) &\ (spectrum_x<spectrum_view_range[0][1])]) expected_maximum_height = spectrum_y_max_in_range - spectrum_view_range[1][0] self.assertAlmostEqual(expected_maximum_height, np.max(line_heights)) def load_phases(self): self.load_phase('ar.jcpds') self.load_phase('ag.jcpds') self.load_phase('au_Anderson.jcpds') self.load_phase('mo.jcpds') self.load_phase('pt.jcpds') self.load_phase('re.jcpds') def load_phase(self, filename): self.controller.add_btn_click_callback(os.path.join(jcpds_path, filename))
class SpectrumModelTest(unittest.TestCase): # TODO: needs to be rewritten to be more in small units etc. def setUp(self): self.x = np.linspace(0.1, 15, 100) self.y = np.sin(self.x) self.spectrum = Spectrum(self.x, self.y) self.spectrum_model = SpectrumModel() def test_set_spectrum(self): self.spectrum_model.set_spectrum(self.x, self.y, 'hoho') assert_array_almost_equal(self.spectrum_model.get_spectrum().x, self.x) assert_array_almost_equal(self.spectrum_model.get_spectrum().y, self.y) self.assertEqual(self.spectrum_model.get_spectrum().name, 'hoho') def test_load_spectrum(self): self.spectrum_model.load_spectrum(os.path.join(data_path, 'spectrum_001.xy')) self.assertEqual(self.spectrum_model.get_spectrum().name, 'spectrum_001') self.assertNotEqual(len(self.x), len(self.spectrum_model.get_spectrum().x)) self.assertNotEqual(len(self.y), len(self.spectrum_model.get_spectrum().y)) def test_add_overlay(self): x_overlay = np.linspace(0, 10) y_overlay = np.linspace(0, 100) self.spectrum_model.add_overlay(x_overlay, y_overlay, "dummy") self.assertEqual(len(self.spectrum_model.overlays), 1) new_overlay = self.spectrum_model.get_overlay(0) self.assertEqual(new_overlay.name, "dummy") assert_array_almost_equal(new_overlay.x, x_overlay) assert_array_almost_equal(new_overlay.y, y_overlay) def test_add_overlay_from_file(self): filename = os.path.join(data_path, 'spectrum_001.xy') self.spectrum_model.add_overlay_file(filename) self.assertEqual(len(self.spectrum_model.overlays), 1) self.assertEqual(self.spectrum_model.get_overlay(0).name, ''.join(os.path.basename(filename).split('.')[0:-1])) def test_add_spectrum_as_overlay(self): self.spectrum_model.add_spectrum_as_overlay() self.assertEqual(len(self.spectrum_model.overlays), 1) assert_array_almost_equal(self.spectrum_model.get_overlay(0).x, self.spectrum_model.spectrum.x) assert_array_almost_equal(self.spectrum_model.get_overlay(0).y, self.spectrum_model.spectrum.y) def test_auto_background_subtraction(self): x = np.linspace(0, 24, 2500) y = np.zeros(x.shape) peaks = [ [10, 3, 0.1], [12, 4, 0.1], [12, 6, 0.1], ] for peak in peaks: y += gaussian(x, peak[0], peak[1], peak[2]) y_bkg = x * 0.4 + 5.0 y_measurement = y + y_bkg self.spectrum_model.set_spectrum(x, y_measurement) auto_background_subtraction_parameters = [2, 50, 50] self.spectrum_model.set_auto_background_subtraction(auto_background_subtraction_parameters) x_spec, y_spec = self.spectrum_model.spectrum.data self.assertAlmostEqual(np.sum(y_spec- y),0)