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) self.converter = Converter()
class TestConverter(TestCase): def setUp(self): TestCase.setUp(self) self.converter = Converter() def tearDown(self): TestCase.tearDown(self) def testskeleton(self): self.assertTrue(True) # def testconvert1(self): # # Base options # ops = Options(name="Test") # ops.beam.energy_eV = 1234 # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # ops.models.add(RANDOM_NUMBER_GENERATOR.press1966_rand1) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(1, len(opss)) # # self.assertAlmostEqual(1234, opss[0].beam.energy_eV, 4) # # self.assertEqual(1, len(opss[0].detectors)) # det = opss[0].detectors['bse'] # self.assertAlmostEqual(0, det.limits_eV[0], 4) # self.assertAlmostEqual(1234, det.limits_eV[1], 4) # self.assertEqual(1000, det.channels) # # self.assertEqual(1, len(opss[0].limits)) # limit = list(opss[0].limits.iterclass(ShowersLimit))[0] # self.assertEqual(5678, limit.showers) # # self.assertEqual(7, len(opss[0].models)) # model = list(opss[0].models.iterclass(RANDOM_NUMBER_GENERATOR))[0] # self.assertEqual(RANDOM_NUMBER_GENERATOR.press1966_rand1, model) # def testconvert2(self): # # Base options # ops = Options(name="Test") # ops.beam = PencilBeam(1234) # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.detectors['photon'] = TransmittedElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # ops.limits.add(TimeLimit(60)) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(1, len(opss)) # # self.assertAlmostEqual(1234, opss[0].beam.energy_eV, 4) # # self.assertEqual(1, len(opss[0].detectors)) # det = opss[0].detectors['bse'] # self.assertAlmostEqual(0, det.limits_eV[0], 4) # self.assertAlmostEqual(1234, det.limits_eV[1], 4) # self.assertEqual(1000, det.channels) # # self.assertEqual(1, len(opss[0].limits)) # limit = list(opss[0].limits.iterclass(ShowersLimit))[0] # self.assertEqual(5678, limit.showers) # # self.assertEqual(7, len(opss[0].models)) # def testconvert3(self): # # Base options # ops = Options(name="Test") # ops.beam.energy_eV = 100e3 # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.detectors['bse2'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(2, len(opss)) def testconvert4(self): # Base options ops = Options(name="Test") ops.beam.energy = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2, 3)) ops.limits.add(ShowersLimit(5678)) # Convert opss = self.converter.convert(ops) # Test self.assertEqual(1, len(opss)) self.assertEqual(2, len(opss[0].detectors)) self.assertEqual(7, len(opss[0].models)) # Test difference in elevation ops.detectors['xray'] = PhotonIntensityDetector((0.5, 1), (2, 3)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # Test difference in azimuth ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2.5, 3)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # Test difference in elevation (PhotonSpectrumDetector) ops.detectors['xray'] = PhotonSpectrumDetector((0.5, 1), (2, 3), 1000, (0, 1234)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # def testconvert6(self): NEW_MODEL_TYPE = ModelType('new') NEW_MODEL_TYPE.test = ('test', ) # Base options ops = Options(name="Test") ops.beam.energy_eV = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) ops.models.add(NEW_MODEL_TYPE.test) ops.limits.add(ShowersLimit(5678)) # Convert opss = self.converter.convert(ops) # Test self.assertEqual(1, len(opss)) # def testconvert7(self): # Base options ops = Options(name="Test") ops.beam.energy_eV = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) # Convert opss = self.converter.convert(ops) # No shower limit self.assertEqual(0, len(opss))
class TestConverter(TestCase): def setUp(self): TestCase.setUp(self) self.converter = Converter() def tearDown(self): TestCase.tearDown(self) def testskeleton(self): self.assertTrue(True) # def testconvert1(self): # # Base options # ops = Options(name="Test") # ops.beam.energy_eV = 1234 # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # ops.models.add(RANDOM_NUMBER_GENERATOR.press1966_rand1) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(1, len(opss)) # # self.assertAlmostEqual(1234, opss[0].beam.energy_eV, 4) # # self.assertEqual(1, len(opss[0].detectors)) # det = opss[0].detectors['bse'] # self.assertAlmostEqual(0, det.limits_eV[0], 4) # self.assertAlmostEqual(1234, det.limits_eV[1], 4) # self.assertEqual(1000, det.channels) # # self.assertEqual(1, len(opss[0].limits)) # limit = list(opss[0].limits.iterclass(ShowersLimit))[0] # self.assertEqual(5678, limit.showers) # # self.assertEqual(7, len(opss[0].models)) # model = list(opss[0].models.iterclass(RANDOM_NUMBER_GENERATOR))[0] # self.assertEqual(RANDOM_NUMBER_GENERATOR.press1966_rand1, model) # def testconvert2(self): # # Base options # ops = Options(name="Test") # ops.beam = PencilBeam(1234) # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.detectors['photon'] = TransmittedElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # ops.limits.add(TimeLimit(60)) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(1, len(opss)) # # self.assertAlmostEqual(1234, opss[0].beam.energy_eV, 4) # # self.assertEqual(1, len(opss[0].detectors)) # det = opss[0].detectors['bse'] # self.assertAlmostEqual(0, det.limits_eV[0], 4) # self.assertAlmostEqual(1234, det.limits_eV[1], 4) # self.assertEqual(1000, det.channels) # # self.assertEqual(1, len(opss[0].limits)) # limit = list(opss[0].limits.iterclass(ShowersLimit))[0] # self.assertEqual(5678, limit.showers) # # self.assertEqual(7, len(opss[0].models)) # def testconvert3(self): # # Base options # ops = Options(name="Test") # ops.beam.energy_eV = 100e3 # ops.detectors['bse'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.detectors['bse2'] = BackscatteredElectronEnergyDetector(1000, (0, 1234)) # ops.limits.add(ShowersLimit(5678)) # # # Convert # opss = self.converter.convert(ops) # # # Test # self.assertEqual(2, len(opss)) def testconvert4(self): # Base options ops = Options(name="Test") ops.beam.energy = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2, 3)) ops.limits.add(ShowersLimit(5678)) # Convert opss = self.converter.convert(ops) # Test self.assertEqual(1, len(opss)) self.assertEqual(2, len(opss[0].detectors)) self.assertEqual(7, len(opss[0].models)) # Test difference in elevation ops.detectors['xray'] = PhotonIntensityDetector((0.5, 1), (2, 3)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # Test difference in azimuth ops.detectors['xray'] = PhotonIntensityDetector((0, 1), (2.5, 3)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # Test difference in elevation (PhotonSpectrumDetector) ops.detectors['xray'] = PhotonSpectrumDetector((0.5, 1), (2, 3), 1000, (0, 1234)) opss = self.converter.convert(ops) self.assertEqual(2, len(opss)) # def testconvert6(self): NEW_MODEL_TYPE = ModelType('new') NEW_MODEL_TYPE.test = ('test',) # Base options ops = Options(name="Test") ops.beam.energy_eV = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) ops.models.add(NEW_MODEL_TYPE.test) ops.limits.add(ShowersLimit(5678)) # Convert opss = self.converter.convert(ops) # Test self.assertEqual(1, len(opss)) # def testconvert7(self): # Base options ops = Options(name="Test") ops.beam.energy_eV = 100e3 ops.detectors['prz'] = PhiZDetector((0, 1), (2, 3), 1000) # Convert opss = self.converter.convert(ops) # No shower limit self.assertEqual(0, len(opss))