Skip to content

test_figure_parser

converter.tests.fluka.test_figure_parser

test_parse_box 

test_parse_box()

Test if box is parsed correctly

Source code in yaptide/converter/tests/fluka/test_figure_parser.py 
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
def test_parse_box():
    """Test if box is parsed correctly"""
    figure = solid_figures.BoxFigure(uuid='ed1507a5-0489-4dc3-bb87-7b27dcff43e7',
                                     name='Box',
                                     position=(2.14, 1.32, 0),
                                     rotation=(0, 0, 0),
                                     x_edge_length=1,
                                     y_edge_length=1,
                                     z_edge_length=1)

    fluka_box = parse_box(figure)

    assert fluka_box.figure_type == "RPP"
    assert fluka_box.uuid == "ed1507a5-0489-4dc3-bb87-7b27dcff43e7"
    assert math.isclose(fluka_box.x_min, 1.64)
    assert math.isclose(fluka_box.x_max, 2.64)
    assert math.isclose(fluka_box.y_min, 0.82)
    assert math.isclose(fluka_box.y_max, 1.82)
    assert math.isclose(fluka_box.z_min, -0.5)
    assert math.isclose(fluka_box.z_max, 0.5)

test_parse_cylinder 

test_parse_cylinder()

Test if cylinder is parsed correctly

Source code in yaptide/converter/tests/fluka/test_figure_parser.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
def test_parse_cylinder():
    """Test if cylinder is parsed correctly"""
    cylinder = solid_figures.CylinderFigure(uuid='ed1507a5-0489-4dc3-bb87-7b27dcff43e7',
                                          name='Cylinder',
                                          position=(2.14, 1.32, 0),
                                          rotation=(19.6, 14.2, 0),
                                          radius_top=1,
                                          radius_bottom=1,
                                          height=1)

    fluka_cylinder = parse_cylinder(cylinder)

    assert fluka_cylinder.figure_type == "RCC"
    assert fluka_cylinder.uuid == "ed1507a5-0489-4dc3-bb87-7b27dcff43e7"
    assert all(math.isclose(a, b, abs_tol=1e-8) for a, b in zip(fluka_cylinder.coordinates,
                                                                (2.0244531744545524, 1.4877257848751275, -0.45663660846935195)))
    assert all(math.isclose(a, b, abs_tol=1e-8) for a, b in zip(fluka_cylinder.height_vector,
                                                                (0.23109365, -0.33545157, 0.91327322)))
    assert fluka_cylinder.radius == 1

test_parse_sphere 

test_parse_sphere()

Test if sphere is parsed correctly

Source code in yaptide/converter/tests/fluka/test_figure_parser.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def test_parse_sphere():
    """Test if sphere is parsed correctly"""
    sphere = solid_figures.SphereFigure(uuid='ed1507a5-0489-4dc3-bb87-7b27dcff43e7',
                                        name='Sphere',
                                        position=(2.14, 1.32, 0),
                                        radius=1)

    fluka_sphere = parse_sphere(sphere)

    assert fluka_sphere.figure_type == "SPH"
    assert fluka_sphere.uuid == "ed1507a5-0489-4dc3-bb87-7b27dcff43e7"
    assert all(math.isclose(a, b, abs_tol=1e-8) for a, b in zip(fluka_sphere.coordinates,
                                                                (2.14, 1.32, 0)))
    assert fluka_sphere.radius == 1

test_rotated_box 

test_rotated_box()

Test if trying to parse rotated box throws an error (as we use RPP for Fluka)

Source code in yaptide/converter/tests/fluka/test_figure_parser.py 
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
def test_rotated_box():
    """
    Test if trying to parse rotated box throws an error
    (as we use RPP for Fluka)
    """
    figure = solid_figures.BoxFigure(uuid='ed1507a5-0489-4dc3-bb87-7b27dcff43e7',
                                     name='Box',
                                     position=(2.14, 1.32, 0),
                                     rotation=(1, 56, 76),
                                     x_edge_length=1,
                                     y_edge_length=1,
                                     z_edge_length=1)

    with pytest.raises(ValueError) as err:
        parse_box(figure)

    assert "Rotation of box is not supported for Fluka" in str(err.value)