class TestConverter(TestCase): def setUp(self): TestCase.setUp(self) self.converter = Converter() def tearDown(self): TestCase.tearDown(self) def testconvert1(self): # Base options ops = Options(name="Test") ops.beam = PencilBeam(1234) ops.detectors['xrays'] = PhotonIntensityDetector((0.1, 0.2), (0.3, 0.4)) ops.limits.add(ShowersLimit(5678)) # Convert with warnings.catch_warnings(record=True) as ws: opss = self.converter.convert(ops) # 6 warnings for the default models self.assertEqual(5, len(ws)) self.assertEqual(1, len(opss)) # Test self.assertAlmostEqual(1234, opss[0].beam.energy_eV, 4) self.assertEqual(1, len(opss[0].detectors)) self.assertEqual(1, len(opss[0].limits)) limit = list(ops.limits.iterclass(ShowersLimit))[0] self.assertEqual(5678, limit.showers) self.assertEqual(5, len(opss[0].models))
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) self.converter = Converter()