ndarray-npy 0.5.0

.npy and .npz file format support for ndarray
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

use ndarray::prelude::*;
use ndarray::{array, Data};
use ndarray_npy::{ReadNpyExt, ReadableElement, WritableElement, WriteNpyExt};
use std::fmt::Debug;

/// Tests that the writing the array as an `.npy` file and reading it
/// back again doesn't change the shape or data.
fn test_round_trip<A, S, D>(before: &ArrayBase<S, D>)
where
    A: Debug + PartialEq + ReadableElement + WritableElement,
    S: Data<Elem = A>,
    D: Dimension,
{
    let mut npy = Vec::<u8>::new();
    before.write_npy(&mut npy).unwrap();
    let after = Array::<A, D>::read_npy(&npy[..]).unwrap();
    assert_eq!(before, &after);
}

#[test]
fn round_trip_i32() {
    test_round_trip(&array![
        [[1i32, 8], [-3, 4], [2, 9]],
        [[-5, 0], [7, 38], [-4, 1]]
    ]);
}

#[test]
fn round_trip_f32() {
    test_round_trip(&array![
        [[3f32, -1.4], [-159., 26.], [5., -3.5]],
        [[-89.7, 93.], [2., 384.], [-626.4, 3.]],
    ]);
}

#[test]
fn round_trip_f64() {
    test_round_trip(&array![
        [2.7, -40.4, -23., 27.8, -49., -43.3],
        [-25.2, 11.8, -8.9, -17.8, 36.4, -25.6],
    ]);
}

#[test]
fn round_trip_bool() {
    test_round_trip(&array![
        [[true], [true], [false]],
        [[false], [true], [false]]
    ]);
}