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])