def setUp(self): unittest.TestCase.setUp(self) self.ops = Options('aatest') self.ops.beam.energy_eV = 4e3 self.ops.geometry.material = \ Material({6: 0.4, 13: 0.6}, absorption_energy_eV={ELECTRON: 234.0}) self.ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2, 3)) self.ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 128) self.ops.limits.add(ShowersLimit(1234)) self.i = Importer() self._testdata = os.path.join(os.path.dirname(__file__), 'testdata')
class TestImporter(unittest.TestCase): def setUp(self): unittest.TestCase.setUp(self) self.ops = Options('aatest') self.ops.beam.energy_eV = 4e3 self.ops.geometry.material = \ Material({6: 0.4, 13: 0.6}, absorption_energy_eV={ELECTRON: 234.0}) self.ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2, 3)) self.ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 128) self.ops.limits.add(ShowersLimit(1234)) self.i = Importer() self._testdata = os.path.join(os.path.dirname(__file__), 'testdata') def tearDown(self): unittest.TestCase.tearDown(self) def testimport_(self): jobdir = os.path.join(self._testdata, 'job1') results = self.i.import_(self.ops, jobdir) self.assertEqual(2, len(results)) result = results['xray'] val, unc = result.intensity('Cu La') self.assertAlmostEqual(3.473295, val, 4) self.assertAlmostEqual(0.0, unc, 4) result = results['prz'] self.assertTrue(result.exists('Au La', True, True)) self.assertTrue(result.exists('Au Ma', True, True)) self.assertEqual((128, 3), result.get('Au La').shape) self.assertEqual((128, 3), result.get('Au Ma').shape)