def notestBackengineWithWPGOut(self):
""" Test that extracting the spectrum from a wpg output works. """
""" notested because requires file not in TestFiles. """
# Construct parameters.
xrts_parameters = self.xrts_parameters
xrts_calculator = PlasmaXRTSCalculator(parameters=xrts_parameters,
input_path='prop_out.h5',
output_path='xrts_out.h5')
# Read in the data.
xrts_calculator._readH5()
# Specify that we want to use the measured source spectrum.
xrts_parameters.source_spectrum = 'prop'
xrts_parameters.energy_range = {
'min': -300.0,
'max': 300.0,
'step': 3.,
}
# Run the backengine.
xrts_calculator.backengine()
self.assertTrue(
'source_spectrum.txt' in os.listdir(xrts_parameters._tmp_dir))
def testReadH5(self):
""" Test the readH5 function to read input from the photon propagator. """
# Construct parameters.
xrts_parameters = self.xrts_parameters
xrts_calculator = PlasmaXRTSCalculator(
parameters=xrts_parameters,
input_path=TestUtilities.generateTestFilePath(
'prop_out_0000001.h5'),
output_path='xrts_out.h5')
xrts_calculator._readH5()
self.assertIn('source_spectrum', xrts_calculator._input_data.keys())
e = xrts_calculator._input_data['source_spectrum'][:, 0]
s = xrts_calculator._input_data['source_spectrum'][:, 1]
def testSerializeSourceSpectrum(self):
""" Test saving the source spectrum to file in the correct location. """
# Construct parameters.
xrts_parameters = self.xrts_parameters
xrts_calculator = PlasmaXRTSCalculator(
parameters=xrts_parameters,
input_path=TestUtilities.generateTestFilePath(
'prop_out_0000001.h5'),
output_path='xrts_out.h5')
xrts_calculator._readH5()
xrts_parameters.source_spectrum = 'prop'
xrts_parameters._serialize()
xrts_calculator._serializeSourceSpectrum()
self.assertTrue(
'source_spectrum.txt' in os.listdir(xrts_parameters._tmp_dir))
def testBackengineWithGauss(self):
""" Test saving the source spectrum to file in the correct location. """
# Construct parameters.
xrts_parameters = self.xrts_parameters
xrts_calculator = PlasmaXRTSCalculator(
parameters=xrts_parameters,
input_path=TestUtilities.generateTestFilePath(
'prop_out_0000001.h5'),
output_path='xrts_out.h5')
# Read in the data.
xrts_calculator._readH5()
# Specify that we want to use the measured source spectrum.
xrts_parameters.source_spectrum_fwhm = 1.0
xrts_parameters.energy_range = {
'min': -200.0,
'max': 200.0,
'step': 2.,
}
# Run the backengine.
xrts_calculator.backengine()
