nalgebra/tests/abomonation.rs

extern crate rand;
extern crate nalgebra;
extern crate abomonation;

use rand::random;
use abomonation::{Abomonation, encode, decode};
use nalgebra::{DMatrix, Matrix3x4, Point3, Translation3, Rotation3};

#[test]
fn abomonate_dmatrix() {
    assert_encode_and_decode(&DMatrix::<f32>::new_random(3, 5));
}

macro_rules! test_abomonation(
    ($($test: ident, $ty: ty);* $(;)*) => {$(
        #[test]
        fn $test() {
            assert_encode_and_decode(&random::<$ty>());
        }
    )*}
);

test_abomonation! {
    abomonate_matrix3x4, Matrix3x4<f32>;
    abomonate_point3, Point3<f32>;
    abomonate_translation3, Translation3<f64>;
    abomonate_rotation3, Rotation3<f64>;
}

fn assert_encode_and_decode<T: Abomonation + PartialEq>(data: &T) {
    let mut bytes = Vec::new();
    unsafe { encode(data, &mut bytes); }

    if let Some((result, rest)) = unsafe { decode::<T>(&mut bytes) } {
        assert!(result == data);
        assert!(rest.len() == 0, "binary data was not decoded completely");
    }
}