numpy 0.12.0

Rust binding of NumPy C-API
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use ndarray::array;
use numpy::{NpyMultiIterBuilder, NpySingleIterBuilder, PyArray};
use pyo3::PyResult;

macro_rules! assert_approx_eq {
    ($x: expr, $y: expr) => {
        assert!(($x - $y) <= std::f64::EPSILON);
    };
}

#[test]
fn readonly_iter() -> PyResult<()> {
    pyo3::Python::with_gil(|py| {
        let data = array![[0.0, 1.0], [2.0, 3.0], [4.0, 5.0]];

        let arr = PyArray::from_array(py, &data);
        let iter = NpySingleIterBuilder::readonly(arr.readonly()).build()?;

        // The order of iteration is not specified, so we should restrict ourselves
        // to tests that don't verify a given order.
        assert_approx_eq!(iter.sum::<f64>(), 15.0);
        Ok(())
    })
}

#[test]
fn mutable_iter() -> PyResult<()> {
    let data = array![[0.0, 1.0], [2.0, 3.0], [4.0, 5.0]];
    pyo3::Python::with_gil(|py| {
        let arr = PyArray::from_array(py, &data);
        let iter = NpySingleIterBuilder::readwrite(arr).build()?;
        for elem in iter {
            *elem *= 2.0;
        }
        let iter = NpySingleIterBuilder::readonly(arr.readonly()).build()?;
        assert_approx_eq!(iter.sum::<f64>(), 30.0);
        Ok(())
    })
}

#[test]
fn multiiter_rr() -> PyResult<()> {
    let data1 = array![[0.0, 1.0], [2.0, 3.0], [4.0, 5.0]];
    let data2 = array![[6.0, 7.0], [8.0, 9.0], [10.0, 11.0]];

    pyo3::Python::with_gil(|py| {
        let arr1 = PyArray::from_array(py, &data1);
        let arr2 = PyArray::from_array(py, &data2);
        let iter = NpyMultiIterBuilder::new()
            .add_readonly(arr1.readonly())
            .add_readonly(arr2.readonly())
            .build()
            .map_err(|e| e.print(py))
            .unwrap();

        let mut sum = 0.0;
        for (x, y) in iter {
            sum += *x * *y;
        }

        assert_approx_eq!(sum, 145.0);
        Ok(())
    })
}

#[test]
fn multiiter_rw() -> PyResult<()> {
    let data1 = array![[0.0, 1.0], [2.0, 3.0], [4.0, 5.0]];
    let data2 = array![[0.0, 0.0], [0.0, 0.0], [0.0, 0.0]];

    pyo3::Python::with_gil(|py| {
        let arr1 = PyArray::from_array(py, &data1);
        let arr2 = PyArray::from_array(py, &data2);
        let iter = NpyMultiIterBuilder::new()
            .add_readonly(arr1.readonly())
            .add_readwrite(arr2)
            .build()?;

        for (x, y) in iter {
            *y = *x * 2.0;
        }

        let iter = NpyMultiIterBuilder::new()
            .add_readonly(arr1.readonly())
            .add_readonly(arr2.readonly())
            .build()?;

        for (x, y) in iter {
            assert_approx_eq!(*x * 2.0, *y);
        }

        Ok(())
    })
}