def test_point_source_mcnp_5():
# Isotropic Source only specifying positon
pt_src = PointSource(1, 2, 3)
exp_str = "SDEF POS=1 2 3\n" \
" ERG=14\n" \
" WGT=1\n" \
" PAR=n"
assert_equal(exp_str, pt_src.mcnp(6))
def test_point_source_mcnp_4():
# Beam Source specifying positon
pt_src = PointSource(1, 2, 3, 4, 5, 6)
exp_str = "SDEF POS=1 2 3\n" \
" VEC=4 5 6 DIR=1\n" \
" ERG=14\n" \
" WGT=1\n" \
" PAR=n"
assert_equal(exp_str, pt_src.mcnp(6))
def test_point_source_mcnp_2():
# Beam Source specifying positon, energy and particle
pt_src = PointSource(1, 2, 3, 4, 5, 6, 12, "Proton")
exp_str = "SDEF POS=1 2 3\n" \
" VEC=4 5 6 DIR=1\n" \
" ERG=12\n" \
" WGT=1\n" \
" PAR=h"
assert_equal(exp_str, pt_src.mcnp(6))
def test_point_source_mcnp_3():
# Beam Source specifying positon and energy
pt_src = PointSource(1, 2, 3, 4, 5, 6, 12)
exp_str = (
"SDEF POS=1 2 3\n"
" VEC=4 5 6 DIR=1\n"
" ERG=12\n"
" WGT=1\n"
" PAR=n"
)
assert_equal(exp_str, pt_src.mcnp(6))
def test_point_source_mcnp_1():
# Beam Source specifying positon, energy, particle and weight
pt_src = PointSource(1, 2, 3, 4, 5, 6, 12, "Proton", 0.5)
exp_str = (
"SDEF POS=1 2 3\n"
" VEC=4 5 6 DIR=1\n"
" ERG=12\n"
" WGT=0.5\n"
" PAR=h"
)
assert_equal(exp_str, pt_src.mcnp(6))
def test_point_source():
pt_src = PointSource(1, 2, 3, 4, 5, 6, 12, "p", 0.5)
test_couples = [(pt_src.x, 1), (pt_src.y, 2), (pt_src.z, 3), (pt_src.u, 4),
(pt_src.v, 5), (pt_src.w, 6), (pt_src.E, 12),
(pt_src.particle, "p"), (pt_src.weight, 0.5)]
for couple in test_couples:
assert_equal(couple[0], couple[1])