def test_storage(self):
p = catima.Projectile(12, 6)
water = catima.get_material(catima.material.Water)
water.thickness(10.0)
graphite = catima.get_material(6)
graphite.thickness(1.0)
data = catima.get_data(p, water)
et = catima.get_energy_table()
self.assertEqual(len(data), 3)
self.assertEqual(len(data[0]), len(et))
res = catima.calculate(p(et[10]), water)
self.assertAlmostEqual(res.range, data[0][10], 6)
self.assertAlmostEqual(catima.range(p, water), data[0][10], 6)
#self.assertAlmostEqual(catima.domega2de(p,water),data[1][10],6)
res = catima.calculate(p(et[100]), water)
self.assertAlmostEqual(res.range, data[0][100], 6)
self.assertAlmostEqual(catima.range(p, water), data[0][100], 6)
#self.assertAlmostEqual(catima.domega2de(p,water),data[1][100],6)
res = catima.calculate(p(et[200]), water)
self.assertAlmostEqual(res.range, data[0][200], 6)
self.assertAlmostEqual(catima.range(p, water), data[0][200], 6)
#self.assertAlmostEqual(catima.domega2de(p,water),data[1][200],6)
res = catima.calculate(p(et[401]), water)
self.assertAlmostEqual(res.range, data[0][401], 6)
self.assertAlmostEqual(catima.range(p, water), data[0][401], 6)
def test_eout_list(self):
graphite = catima.get_material(6)
graphite.thickness(0.5)
p = catima.Projectile(12, 6)
energies = [100, 500, 1000]
res = catima.calculate(p(1000), graphite)
self.assertAlmostEqual(res.Eout, 997.077, 1)
res2 = catima.energy_out(p, energies, graphite)
self.assertEqual(len(res2), len(energies))
self.assertAlmostEqual(res2[2], 997.077, 1)
self.assertAlmostEqual(res2[0],
catima.calculate(p(energies[0]), graphite).Eout,
1)
self.assertAlmostEqual(res2[1],
catima.calculate(p(energies[1]), graphite).Eout,
1)
def test_config(self):
water = catima.get_material(catima.material.Water)
water.density(1.0)
water.thickness(1.0)
p = catima.Projectile(1, 1)
conf = catima.Config()
conf.calculation = catima.omega_types.bohr
conf2 = catima.Config()
conf2.calculation = catima.omega_types.atima
self.assertEqual(conf.calculation, 1)
self.assertEqual(conf2.calculation, 0)
p(1000)
res = catima.calculate(p, water, config=conf)
res2 = catima.calculate(p, water, config=conf2)
self.assertAlmostEqual(res.dEdxi, res2.dEdxi, delta=1e-6)
self.assertNotAlmostEqual(res.sigma_E, res2.sigma_E, delta=1e-4)
self.assertNotAlmostEqual(res.sigma_r, res2.sigma_r, delta=1e-4)
def test_eout(self):
graphite = catima.get_material(6)
graphite.thickness(0.5)
p = catima.Projectile(12, 6)
res = catima.calculate(p(1000), graphite)
res2 = catima.energy_out(p(1000), graphite)
self.assertAlmostEqual(res.Eout, 997.077, 1)
self.assertAlmostEqual(res.Eout, res2, 3)
def test_material_calculation(self):
water = catima.get_material(catima.material.Water)
p = catima.Projectile(1, 1)
p(1000)
res = catima.calculate(p, water)
res2 = catima.dedx(p, water)
self.assertAlmostEqual(res.dEdxi, 2.23, 1)
self.assertAlmostEqual(res.dEdxi, res2, 3)
res = catima.calculate(p(500), water)
res2 = catima.dedx(p, water)
self.assertAlmostEqual(res.dEdxi, 2.76, 1)
self.assertAlmostEqual(res.dEdxi, res2, 3)
res = catima.calculate(p(9), water)
res2 = catima.dedx(p, water)
self.assertAlmostEqual(res.dEdxi, 51.17, 1)
self.assertAlmostEqual(res.dEdxi, res2, 3)
res = catima.calculate(p(9), water)
res = catima.calculate(p(9), water)
self.assertAlmostEqual(res.dEdxi, 51.17, 1)
p(900000)
res = catima.calculate(p, water)
res2 = catima.dedx_from_range(p, water)
self.assertAlmostEqual(res.dEdxi, res2, 3)