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

use cervo_core::prelude::{EpsilonInjector, Inferer, InfererExt};

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

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

    let shapes = instance.input_shapes().to_vec();
    let observations = helpers::build_inputs_from_desc(1, &shapes);

    let result = instance.infer_batch(observations);
    assert!(result.is_ok());

    let result = result.unwrap();
    assert_eq!(result.len(), 1);
    assert_eq!(result[&0].data["Identity"].len(), 1);
}