// Author: Tom Solberg <tom.solberg@embark-studios.com>
// Copyright © 2022, Embark Studios AB, all rights reserved.
// Created: 10 May 2022

use cervo_core::prelude::Inferer;

#[path = "./helpers.rs"]
mod helpers;

#[test]
fn test_load_onnx_simple() {
    let mut reader = helpers::get_file("test.onnx").unwrap();
    cervo_onnx::builder(&mut reader)
        .build_basic()
        .expect("loading success");
}

#[test]
fn test_load_input_shape_simple() {
    let mut reader = helpers::get_file("test.onnx").unwrap();
    let instance = cervo_onnx::builder(&mut reader)
        .build_basic()
        .expect("failed reading instance");
    assert_eq!(
        instance.input_shapes()[0].1,
        [2],
        "mismatch in input shapes"
    );
}

#[test]
fn test_load_output_shape_simple() {
    let mut reader = helpers::get_file("test.onnx").unwrap();
    let instance = cervo_onnx::builder(&mut reader).build_basic().unwrap();

    assert_eq!(
        instance.output_shapes()[0].1,
        &[1],
        "mismatch in output shapes",
    );
}