neopdf_pyapi 0.3.0-alpha2

Python bindings to NeoPDF Rust library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

import numpy as np

from neopdf.gridpdf import SubGrid, GridArray


class TestGridPDF:
    def test_subgrid(self, xq2_points):
        xmin, xmax, q2min, q2max = (1e-5, 1.0, 1.65, 1.0e8)
        xs, q2s = xq2_points(xmin, xmax, q2min, q2max)
        kts = [0.5, 1.0]
        xis = [0.0]  # dummy values
        deltas = [0.0]  # dummy values
        nucleons = [1.0, 2.0]
        alphas = [0.118, 0.120]
        grid = np.random.rand(
            len(nucleons),
            len(alphas),
            len(xis),
            len(deltas),
            len(kts),
            len(xs),
            len(q2s),
            1,
        )

        subgrid = SubGrid(xs, q2s, kts, xis, deltas, nucleons, alphas, grid)

        assert subgrid.alphas_range() == (0.118, 0.120)
        assert subgrid.x_range() == (xs[0], xs[-1])
        assert subgrid.q2_range() == (q2s[0], q2s[-1])
        assert subgrid.grid_shape() == [
            len(nucleons),
            len(alphas),
            len(xis),
            len(deltas),
            len(kts),
            len(xs),
            len(q2s),
            1,
        ]

    def test_gridarray(self, xq2_points):
        xmin, xmax, q2min, q2max = (1e-5, 1.0, 1.65, 1.0e8)
        xs, q2s = xq2_points(xmin, xmax, q2min, q2max)
        kts = [0.5, 1.0]
        xis = [0.0]  # dummy values
        deltas = [0.0]  # dummy values
        nucleons = [1.0, 2.0]
        alphas = [0.118, 0.120]
        grid = np.random.rand(
            len(nucleons),
            len(alphas),
            len(xis),
            len(deltas),
            len(kts),
            len(xs),
            len(q2s),
            1,
        )

        subgrid1 = SubGrid(xs, q2s, kts, xis, deltas, nucleons, alphas, grid)
        subgrid2 = SubGrid(xs, q2s, kts, xis, deltas, nucleons, alphas, grid)

        pids = [21, -2, -1, 1, 2]
        grid_array = GridArray(pids, [subgrid1, subgrid2])

        assert grid_array.pids() == pids
        assert len(grid_array.subgrids()) == 2