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