interpn 0.11.0

N-dimensional interpolation/extrapolation methods, no-std and no-alloc compatible.
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

import numpy as np
import pytest

from interpn import interpn


@pytest.mark.parametrize("dtype", [np.float64, np.float32])
@pytest.mark.parametrize("max_threads", [None, 1], ids=["parallel", "serial"])
def test_interpn_check_bounds_regular(dtype, max_threads):
    grid = np.linspace(-1.0, 1.0, 5).astype(dtype)
    vals = np.linspace(0.0, 10.0, grid.size).astype(dtype)

    obs_inside = [np.array([-0.5, 0.5], dtype=dtype)]
    obs_outside = [np.array([-0.5, 1.5], dtype=dtype)]

    inside = interpn(
        obs=obs_inside,
        grids=[grid],
        vals=vals,
        method="linear",
        check_bounds_with_atol=1e-8,
        max_threads=max_threads,
    )
    assert inside.shape == obs_inside[0].shape

    with pytest.raises(AssertionError):
        interpn(
            obs=obs_outside,
            grids=[grid],
            vals=vals,
            method="linear",
            check_bounds_with_atol=1e-8,
            max_threads=max_threads,
        )


@pytest.mark.parametrize("dtype", [np.float64, np.float32])
@pytest.mark.parametrize("max_threads", [None, 1], ids=["parallel", "serial"])
def test_interpn_check_bounds_rectilinear(dtype, max_threads):
    grid = np.array([-1.0, -0.25, 0.5, 2.0], dtype=dtype)
    vals = np.linspace(0.0, 10.0, grid.size).astype(dtype)

    obs_inside = [np.array([-0.5, 1.0], dtype=dtype)]
    obs_outside = [np.array([-1.5, 0.25], dtype=dtype)]

    inside = interpn(
        obs=obs_inside,
        grids=[grid],
        vals=vals,
        method="linear",
        check_bounds_with_atol=1e-8,
        max_threads=max_threads,
    )
    assert inside.shape == obs_inside[0].shape

    with pytest.raises(AssertionError):
        interpn(
            obs=obs_outside,
            grids=[grid],
            vals=vals,
            method="linear",
            check_bounds_with_atol=1e-8,
            max_threads=max_threads,
        )