Skip to content

test_fluka_beam

converter.tests.fluka.test_fluka_beam

test_parse_fluka_beam 

test_parse_fluka_beam(project_fluka_json)

Test if Fluka beam is parsed correctly

Source code in yaptide/converter/tests/fluka/test_fluka_beam.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
def test_parse_fluka_beam(project_fluka_json):
    """Test if Fluka beam is parsed correctly"""
    beam_json = project_fluka_json['beam']

    fluka_beam = parse_beam(beam_json)

    assert fluka_beam.energy_MeV == 70
    assert fluka_beam.particle_name == 'PROTON'
    assert fluka_beam.shape == BeamShape.CIRCULAR
    assert fluka_beam.shape_x == 0
    assert fluka_beam.shape_y == 3
    assert fluka_beam.z_negative is False
    assert fluka_beam.beam_pos == (0, 0, -1.5)
    assert isclose(fluka_beam.beam_dir[0], 0.0, abs_tol=1e-16)
    assert isclose(fluka_beam.beam_dir[1], 0.0, abs_tol=1e-16)

test_parse_heavy_ions 

test_parse_heavy_ions(project_fluka_json)

Test if Fluka beam is parsed correctly

Source code in yaptide/converter/tests/fluka/test_fluka_beam.py 
23
24
25
26
27
28
29
30
31
32
33
34
35
def test_parse_heavy_ions(project_fluka_json):
    """Test if Fluka beam is parsed correctly"""
    beam_json = copy.deepcopy(project_fluka_json['beam'])
    beam_json['particle']['id'] = 25
    beam_json['particle']['a'] = 6
    beam_json['particle']['z'] = 12

    fluka_beam = parse_beam(beam_json)

    assert fluka_beam.energy_MeV == 70
    assert fluka_beam.particle_name == 'HEAVYION'
    assert fluka_beam.heavy_ion_a == 6
    assert fluka_beam.heavy_ion_z == 12