def test_map_background_subtracted_signal_keeps_input_dtype(self):
si_xdata_32 = self.__create_spectrum_image_xdata(dtype=numpy.float32)
si_xdata_64 = self.__create_spectrum_image_xdata(dtype=numpy.float64)
mapped_xdata_32 = eels_analysis.map_background_subtracted_signal(
si_xdata_32, None, [(0.2, 0.3)], (0.4, 0.5))
mapped_xdata_64 = eels_analysis.map_background_subtracted_signal(
si_xdata_64, None, [(0.2, 0.3)], (0.4, 0.5))
self.assertEqual(numpy.float32, mapped_xdata_32.data.dtype)
self.assertEqual(numpy.float64, mapped_xdata_64.data.dtype)
def execute(self, spectrum_image_data_item, fit_interval_graphics,
signal_interval_graphic):
spectrum_image_xdata = spectrum_image_data_item.xdata
fit_intervals = [
fit_interval_graphic.interval
for fit_interval_graphic in fit_interval_graphics
]
signal_interval = signal_interval_graphic.interval
self.__mapped_xdata = eels_analysis.map_background_subtracted_signal(
spectrum_image_xdata, None, fit_intervals, signal_interval)
def execute(self, **kwargs):
spectrum_image_xdata = kwargs["spectrum_image_xdata"]
fit_interval = kwargs["fit_interval"]
signal_interval = kwargs["signal_interval"]
atomic_number = kwargs.get("atomic_number")
shell_number = kwargs.get("shell_number")
subshell_index = kwargs.get("subshell_index")
electron_shell = None
if atomic_number is not None and shell_number is not None and subshell_index is not None:
electron_shell = PeriodicTable.ElectronShell(atomic_number, shell_number, subshell_index)
self.__mapped_xdata = eels_analysis.map_background_subtracted_signal(spectrum_image_xdata, electron_shell, [fit_interval], signal_interval)
def test_map_background_subtracted_signal_produces_correct_calibrations(
self):
calibration = Calibration.Calibration(200.0, 2.0, 'eV')
calibration_y = Calibration.Calibration(101.0, 1.5, 'nm')
calibration_x = Calibration.Calibration(102.0, 2.5, 'nm')
spectrum_length = 1000
w, h = 20, 20
electron_shell = PeriodicTable.ElectronShell(1, 1, 0)
data_and_metadata = DataAndMetadata.DataAndMetadata.from_data(
numpy.ones((spectrum_length, w, h), numpy.float),
dimensional_calibrations=[
calibration_y, calibration_x, calibration
])
mapped = eels_analysis.map_background_subtracted_signal(
data_and_metadata, electron_shell, [(0.2, 0.3)], (0.4, 0.5))
self.assertEqual(len(mapped.dimensional_shape), 2)
self.assertEqual(len(mapped.dimensional_calibrations), 2)
self.assertEqual(mapped.dimensional_calibrations[0], calibration_y)
self.assertEqual(mapped.dimensional_calibrations[1], calibration_x)
def execute(self, spectrum_image_xdata, fit_interval, signal_interval):
self.__mapped_xdata = eels_analysis.map_background_subtracted_signal(spectrum_image_xdata, None, [fit_interval], signal_interval)
