Skip to content

test_shieldhit_beam

converter.tests.shieldhit.test_shieldhit_beam

test_generated_beam_dat

test_generated_beam_dat(
    project_shieldhit_json, sh12a_parser, tmpdir
)

Check if beam.dat file created properly

Source code in yaptide/converter/tests/shieldhit/test_shieldhit_beam.py
 5
 6
 7
 8
 9
10
11
12
13
14
15
def test_generated_beam_dat(project_shieldhit_json, sh12a_parser, tmpdir) -> None:
    """Check if beam.dat file created properly"""
    output_dir = Path(tmpdir)
    run_parser(sh12a_parser, project_shieldhit_json, output_dir)
    with open(output_dir / 'beam.dat') as f:
        input_text = f.read()
        assert input_text
        assert "JPART0       	2" in input_text
        assert "STRAGG          2" in input_text
        assert "MSCAT           2" in input_text
        assert "NUCRE           1" in input_text