def setUp(self):
TestCase.setUp(self)
# Results 1
self.ops1 = Options(name='test1')
self.ops1.detectors['det1'] = PhotonIntensityDetector((0, 1), (0, 1))
self.ops1.detectors['det2'] = TimeDetector()
self.ops1.detectors['det3'] = ElectronFractionDetector()
results1 = {}
results1['det1'] = PhotonIntensityResult()
results1['det2'] = TimeResult()
results1['det3'] = ElectronFractionResult()
# Results 2
ops2 = Options(name='test2')
ops2.detectors['det1'] = PhotonIntensityDetector((0, 1), (0, 1))
results2 = {}
results2['det1'] = PhotonIntensityResult()
# Base options
self.ops = Options(name='base')
# Sequence
list_results = [(self.ops1, results1), (ops2, results2)]
self.results = Results(self.ops, list_results)
def setUp(self):
TestCase.setUp(self)
self.t1 = Transition(29, 9, 4)
distributions = {}
gnf_zs = [1.0, 2.0, 3.0, 4.0]
gnf_values = [0.0, 5.0, 4.0, 1.0]
gnf_uncs = [0.01, 0.02, 0.03, 0.04]
gnf = np.array([gnf_zs, gnf_values, gnf_uncs]).T
distributions[PhotonKey(self.t1, False, PhotonKey.P)] = gnf
gt_zs = [1.0, 2.0, 3.0, 4.0]
gt_values = [10.0, 15.0, 14.0, 11.0]
gt_uncs = [0.11, 0.12, 0.13, 0.14]
gt = np.array([gt_zs, gt_values, gt_uncs]).T
distributions[PhotonKey(self.t1, False, PhotonKey.T)] = gt
enf_zs = [1.0, 2.0, 3.0, 4.0]
enf_values = [20.0, 25.0, 24.0, 21.0]
enf = np.array([enf_zs, enf_values]).T
distributions[PhotonKey(self.t1, True, PhotonKey.P)] = enf
et_zs = [1.0, 2.0, 3.0, 4.0]
et_values = [30.0, 35.0, 34.0, 31.0]
et_uncs = [0.31, 0.32, 0.33, 0.34]
et = np.array([et_zs, et_values, et_uncs]).T
distributions[PhotonKey(self.t1, True, PhotonKey.T)] = et
self.r = PhiZResult(distributions)
def setUp(self):
TestCase.setUp(self)
self.geo = PenelopeGeometry('Test Geometry')
surface1 = zplane(1e-10)
surface2 = zplane(-1e-3)
surface3 = cylinder(1e-2)
surface4 = xplane(0.0)
mat1 = PenelopeMaterial.pure(29)
self.module1 = Module(self.geo, mat1)
self.module1.add_surface(surface1, SIDEPOINTER_NEGATIVE)
self.module1.add_surface(surface2, SIDEPOINTER_POSITIVE)
self.module1.add_surface(surface3, SIDEPOINTER_NEGATIVE)
self.module1.add_surface(surface4, SIDEPOINTER_POSITIVE)
self.geo.modules.add(self.module1)
mat2 = PenelopeMaterial.pure(30)
self.module2 = Module(self.geo, mat2)
self.module2.add_surface(surface1, SIDEPOINTER_NEGATIVE)
self.module2.add_surface(surface2, SIDEPOINTER_POSITIVE)
self.module2.add_surface(surface3, SIDEPOINTER_NEGATIVE)
self.module2.add_module(self.module1)
self.geo.modules.add(self.module2)
self.geo.tilt_rad = radians(45)
def setUp(self):
TestCase.setUp(self)
intforce = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION, -4)
mat1 = PenelopeMaterial.pure(29, interaction_forcings=[intforce],
maximum_step_length_m=1e4)
mat1._index = 1
mat2 = VACUUM
mat2._index = 0
surface1 = SurfaceImplicit()
surface1._index = 0
surface2 = SurfaceImplicit()
surface2._index = 1
self.module2 = Module(None, mat2)
self.module2._index = 1
self.module1 = Module(None, mat1, 'Test')
self.module1.add_surface(surface1, -1)
self.module1.add_surface(surface2, 1)
self.module1.add_module(self.module2)
self.module1.rotation.phi_rad = radians(180)
self.module1.shift.z_m = -1e3
self.module1._index = 0
self.materials_lookup = {0: mat2, 1: mat1}
self.surfaces_lookup = {0: surface1, 1: surface2}
self.modules_lookup = {0: self.module1, 1: self.module2}
def setUp(self):
TestCase.setUp(self)
experimental_kratios = {29: (0.2, 0.0), 79: (0.8, 0.0)}
initial_composition = {29: 0.5, 79: 0.5}
self.it = Wegstein1958Iterator(experimental_kratios, initial_composition)
def setUp(self):
TestCase.setUp(self)
self.tmpdir = tempfile.mkdtemp()
self.testdata = os.path.join(os.path.dirname(__file__),
'../../testdata')
self.updater = Updater()
def setUp(self):
TestCase.setUp(self)
self.tmpdir = tempfile.mkdtemp()
self.c = Converter()
self.e = Exporter()
def setUp(self):
TestCase.setUp(self)
self.tmpdir = tempfile.mkdtemp()
self.c = Converter()
self.e = Exporter()
def setUp(self):
TestCase.setUp(self)
self.geo = PenelopeGeometry('Test Geometry')
surface1 = zplane(1e-10)
surface2 = zplane(-1e-3)
surface3 = cylinder(1e-2)
surface4 = xplane(0.0)
mat1 = PenelopeMaterial.pure(29)
self.module1 = Module(self.geo, mat1)
self.module1.add_surface(surface1, SIDEPOINTER_NEGATIVE)
self.module1.add_surface(surface2, SIDEPOINTER_POSITIVE)
self.module1.add_surface(surface3, SIDEPOINTER_NEGATIVE)
self.module1.add_surface(surface4, SIDEPOINTER_POSITIVE)
self.geo.modules.add(self.module1)
mat2 = PenelopeMaterial.pure(30)
self.module2 = Module(self.geo, mat2)
self.module2.add_surface(surface1, SIDEPOINTER_NEGATIVE)
self.module2.add_surface(surface2, SIDEPOINTER_POSITIVE)
self.module2.add_surface(surface3, SIDEPOINTER_NEGATIVE)
self.module2.add_module(self.module1)
self.geo.modules.add(self.module2)
self.geo.tilt_rad = radians(45)
def setUp(self):
TestCase.setUp(self)
self.testdata = os.path.join(os.path.dirname(__file__),
'testdata', 'test1')
self.i = Importer()
def setUp(self):
TestCase.setUp(self)
intforce = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION,
-4)
mat1 = PenelopeMaterial.pure(29,
interaction_forcings=[intforce],
maximum_step_length_m=1e4)
mat1._index = 1
mat2 = VACUUM
mat2._index = 0
surface1 = SurfaceImplicit()
surface1._index = 0
surface2 = SurfaceImplicit()
surface2._index = 1
self.module2 = Module(None, mat2)
self.module2._index = 1
self.module1 = Module(None, mat1, 'Test')
self.module1.add_surface(surface1, -1)
self.module1.add_surface(surface2, 1)
self.module1.add_module(self.module2)
self.module1.rotation.phi_rad = radians(180)
self.module1.shift.z_m = -1e3
self.module1._index = 0
self.materials_lookup = {0: mat2, 1: mat1}
self.surfaces_lookup = {0: surface1, 1: surface2}
self.modules_lookup = {0: self.module1, 1: self.module2}
def setUp(self):
TestCase.setUp(self)
experimental_kratios = {29: (0.2, 0.0), 79: (0.8, 0.0)}
initial_composition = {29: 0.5, 79: 0.5}
self.it = SimpleIterator(experimental_kratios, initial_composition)
def setUp(self):
TestCase.setUp(self)
self.tmpdir = tempfile.mkdtemp()
self.testdata = os.path.join(os.path.dirname(__file__),
'../../testdata')
self.updater = Updater()
def setUp(self):
TestCase.setUp(self)
baseops = Options()
baseops.detectors["xray"] = PhotonIntensityDetector((0, 1), (0, 3))
self.m = Measurement(baseops)
self.m.add_kratio(Ka(29), 0.2470, 0.004)
def setUp(self):
TestCase.setUp(self)
coefficients = (1e3, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1e9, 0.0, 0.0)
self.surface = SurfaceImplicit(coefficients, description='surface')
self.surface.rotation.phi_rad = radians(180)
self.surface.shift.z_m = -1e3
self.surface._index = 0
def setUp(self):
TestCase.setUp(self)
coefficients = (1e3, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1e9, 0.0, 0.0)
self.surface = SurfaceImplicit(coefficients, description='surface')
self.surface.rotation.phi_rad = radians(180)
self.surface.shift.z_m = -1e3
self.surface._index = 0
def setUp(self):
TestCase.setUp(self)
self.tmpdir = tempfile.mkdtemp()
self.ops = Options("test1")
self.e = XMLExporter(DummyConverter)
def setUp(self):
TestCase.setUp(self)
self.ops = Options(name="Test")
self.ops.beam.energy = 20e3
mat = Material({29: 0.5, 30: 0.4, 50: 0.1}, 'Brass')
self.ops.geometry.material = mat
def setUp(self):
TestCase.setUp(self)
baseops = Options()
baseops.detectors['xray'] = PhotonIntensityDetector((0, 1), (0, 3))
self.m = Measurement(baseops)
self.m.add_kratio(Ka(29), 0.2470, 0.004)
def setUp(self):
TestCase.setUp(self)
lines = ['[section1]', 'option1=value1', 'option2=value2',
'[section2]', 'option3=value3', 'option4=value4']
sobj = StringIO('\n'.join(lines))
self.c = ConfigParser()
self.c.read(sobj)
def setUp(self):
TestCase.setUp(self)
self.t1 = Transition(29, 9, 4)
distributions = {}
enf = np.ones((5, 5, 5))
distributions[PhotonKey(self.t1, True, PhotonKey.P)] = enf
self.r = PhotonEmissionMapResult(distributions)
def setUp(self):
TestCase.setUp(self)
self.if1 = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION, -4, (0.1, 1.0))
self.if2 = InteractionForcing(ELECTRON, INNERSHELL_IMPACT_IONISATION, -400, (0.1, 1.0))
self.if3 = InteractionForcing(PHOTON, INCOHERENT_COMPTON_SCATTERING, -10, (1e-4, 1.0))
self.if4 = InteractionForcing(PHOTON, INCOHERENT_COMPTON_SCATTERING, -10, (1e-4, 1.0))
self.if5 = InteractionForcing(POSITRON, ANNIHILATION, -100, (1e-4, 1.0))
self.if6 = InteractionForcing(POSITRON, DELTA, -100, (1e-4, 1.0))
def setUp(self):
TestCase.setUp(self)
if1 = InteractionForcing(ELECTRON, HARD_BREMSSTRAHLUNG_EMISSION,
- 4, (0.1, 1.0))
self.m = PenelopeMaterial({'Cu': 1.0}, 'Pure Cu', density_kg_m3=8960.0,
elastic_scattering=(0.1, 0.2),
cutoff_energy_inelastic_eV=51.2,
cutoff_energy_bremsstrahlung_eV=53.4,
interaction_forcings=[if1],
maximum_step_length_m=123.456)
def setUp(self):
TestCase.setUp(self)
self.ops = Options(name="Test")
self.ops.programs.add(DummyProgram())
self.ops.beam.energy_eV = 1234
self.ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234))
self.ops.limits.add(ShowersLimit(5678))
self.ops.models.add(ELASTIC_CROSS_SECTION.rutherford)
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.detectors['traj'] = TrajectoryDetector(False)
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options("test")
ops.detectors["time"] = TimeDetector()
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.geometry.body.material = Material.pure(79)
ops.detectors['time'] = TimeDetector()
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
energies_eV = [1.0, 1.5, 2.0, 2.5]
total_val = [6.0, 9.0, 1.0, 5.0]
total_unc = [0.1, 0.5, 0.9, 0.05]
background_val = [1.0, 2.0, 2.0, 0.5]
background_unc = [0.05, 0.04, 0.03, 0.02]
total = np.array([energies_eV, total_val, total_unc]).T
background = np.array([energies_eV, background_val, background_unc]).T
self.r = PhotonSpectrumResult(total, background)
def setUp(self):
TestCase.setUp(self)
options = Options('PAP')
options.beam.energy_eV = 20000
detector = PhotonIntensityDetector((radians(50.0), radians(55)),
(0.0, radians(360.0)))
self.m = Measurement(options, options.geometry.body, detector)
self.m.add_kratio(Ka(29), 0.2470, 0.004)
self.m.add_rule(ElementByDifferenceRule(79))
def setUp(self):
TestCase.setUp(self)
options = Options('PAP')
options.beam.energy_eV = 20000
detector = PhotonIntensityDetector((radians(50.0), radians(55)),
(0.0, radians(360.0)))
self.m = Measurement(options, options.geometry.body, detector)
self.m.add_kratio(Ka(29), 0.2470, 0.004)
self.m.add_rule(ElementByDifferenceRule(79))
def setUp(self):
TestCase.setUp(self)
self.g1 = VerticalLayers(Material.pure(29), Material.pure(30))
self.g1.add_layer(Material.pure(31), 500.0)
self.g2 = VerticalLayers(Material.pure(29), Material.pure(30))
self.g2.add_layer(Material.pure(29), 100.0)
self.g2.add_layer(Material.pure(32), 200.0)
self.g3 = VerticalLayers(Material.pure(29), Material.pure(30))
self.g3.add_layer(Material.pure(31), 500.0)
self.g3.depth_m = 400.0
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.beam.energy_keV = 10
ops.geometry.body.material = Material.pure(29)
ops.detectors['fraction'] = ElectronFractionDetector()
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.beam.origin_m = (0.0, 0.0, 0.001)
ops.geometry.body.material = \
Material({79: 1.0}, absorption_energy_eV={ELECTRON: 56.0})
ops.detectors['time'] = TimeDetector()
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
# Create test files
self.tmpdir = tempfile.mkdtemp()
self.invalid_filepath = os.path.join(self.tmpdir, 'test1.txt')
with open(self.invalid_filepath, 'w') as fp:
fp.write('Version 0 file, invalid')
self.valid_filepath = os.path.join(self.tmpdir, 'test2.txt')
with open(self.valid_filepath, 'w') as fp:
fp.write('Hello')
# Updater
self.updater = MockUpdater()
def setUp(self):
TestCase.setUp(self)
self.g1 = HorizontalLayers(Material.pure(29))
self.g2 = HorizontalLayers(None) # No substrate
self.g3 = HorizontalLayers(Material.pure(29)) # Empty layer
self.g1.add_layer(Material.pure(30), 123.456)
self.g1.add_layer(Material.pure(31), 456.789)
self.g2.add_layer(Material.pure(30), 123.456)
self.g2.add_layer(Material.pure(31), 456.789)
self.g3.add_layer(Material.pure(30), 123.456)
self.g3.add_layer(Material.pure(31), 456.789)
self.g3.add_layer(VACUUM, 456.123)
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.beam = PencilBeam(5e3)
ops.geometry.body.material = \
Material({6: 0.4, 13: 0.6}, absorption_energy_eV={ELECTRON: 234.0})
ops.detectors['xray'] = \
PhotonIntensityDetector.annular(d2r(40.0), d2r(5.0))
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
geometry = MultiLayers(pure(29))
geometry.add_layer(Material("AuPd", {79: 0.5, 46: 0.5}), 100e-9)
baseops = Options()
baseops.detectors['xray'] = PhotonIntensityDetector((0, 1), (0, 3))
meas = Measurement(baseops)
meas.add_kratio(Ka(29), 0.2470, 0.004)
getter = lambda geometry: geometry.layers[0].thickness_m
setter = lambda geometry, val: setattr(geometry.layers[0], 'thickness_m', val)
param = Parameter(getter, setter, 100e-9, 10e-9, 500e-9)
self.exp = Experiment(geometry, [meas], [param])
def setUp(self):
TestCase.setUp(self)
self.outputdir = tempfile.mkdtemp()
self.workdir = tempfile.mkdtemp()
ops = Options('test')
ops.beam = PencilBeam(5e3)
ops.geometry.body.material = \
Material({6: 0.4, 13: 0.6}, absorption_energy_eV={ELECTRON: 234.0})
ops.detectors['xray'] = \
PhotonIntensityDetector.annular(d2r(40.0), d2r(5.0))
ops.limits.add(ShowersLimit(1))
self.ops = Converter().convert(ops)[0]
self.worker = Worker(program)
def setUp(self):
TestCase.setUp(self)
# Options
self.ops = Options()
self.ops.detectors['det1'] = PhotonIntensityDetector((0, 1), (0, 1))
self.ops.detectors['det2'] = TimeDetector()
self.ops.detectors['det3'] = ElectronFractionDetector()
# Results
results = {}
results['det1'] = PhotonIntensityResult()
results['det2'] = TimeResult()
results['det3'] = ElectronFractionResult()
self.results = ResultsContainer(self.ops, results)
def setUp(self):
TestCase.setUp(self)
self.ops = Options()
self.ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2, 3))
self.ops.detectors['fraction'] = ElectronFractionDetector()
self.ops.detectors['phiz'] = PhiZDetector((0, 1), (2, 3), 500)
self.ops.detectors['photonradial'] = PhotonRadialDetector((0, 1), (2, 3), 500)
self.ops.detectors['bseenergy'] = BackscatteredElectronEnergyDetector(500, (0, 30e3))
self.ops.detectors['teenergy'] = TransmittedElectronEnergyDetector(500, (0, 30e3))
self.ops.detectors['bseangle'] = BackscatteredElectronPolarAngularDetector(91)
self.ops.detectors['bseradial'] = BackscatteredElectronRadialDetector(500)
self.ops.detectors['trajs'] = TrajectoryDetector()
filepath = os.path.join(os.path.dirname(__file__),
'testdata', 'result1.cas')
imp = Importer()
with open(filepath, 'rb') as f:
self.results = imp.import_cas(self.ops, f)
def setUp(self):
TestCase.setUp(self)
self.scale = Scale(1.0, 2.0, 3.0)
def setUp(self):
TestCase.setUp(self)
self.rule = ElementByDifferenceRule(79)
def setUp(self):
TestCase.setUp(self)
self.e = Exporter()
def tearDown(self):
TestCase.tearDown(self)
shutil.rmtree(self.tmpdir, ignore_errors=True)
