def test_dark_correction(self): desired_image_data = copy.copy(self.msi.get_image()) desired_image_data -= 1 dark = copy.copy(self.msi) dark.set_image(np.ones_like(dark.get_image())) mani.dark_correction(self.msi, dark) np.testing.assert_equal(self.msi.get_image(), desired_image_data, "dark image correctly accounted for") np.testing.assert_equal(dark.get_image(), np.ones_like(dark.get_image()), "dark image unchanged by dark correction")
def test_dark_correction_with_single_value(self): desired_image_data = copy.copy(self.specialmsi.get_image()) desired_image_data -= 1 dark = copy.copy(self.specialmsi) dark.set_image(np.ones_like(dark.get_image())) mani.calculate_mean_spectrum(dark) mani.dark_correction(self.specialmsi, dark) np.testing.assert_equal(self.specialmsi.get_image(), desired_image_data, "dark image correctly accounted for from singular dark value") np.testing.assert_equal(dark.get_image(), np.ones_like(dark.get_image()), "dark image unchanged by dark correction")