You are an expert in writing code in many different languages. Your goal is to perform code completion for the following code, keeping in mind the rest of the code and the file meta data.
    Metadata details:
    filename: identity.rs
    path of file: ./repos/wonnx/wonnx/tests
    
    the code of the file until where you have to start completion: use std::{collections::HashMap, convert::TryInto};
use wonnx::utils::{graph, model, node, tensor, OutputTensor};
mod common;

#[test]
fn test_identity() {
    let n: usize = 16;
    let mut input_data = HashMap::new();

    let data: Vec<f32> = (0..n).map(|x| x as f32).collect();
    let dims = vec![n as i64];
    input_data.insert("X".to_string(), data.as_slice().into());

    // Model: X -> Identity -> Y; Y==Z
    let model = model(graph(
        vec![tensor("X", &dims)],
        vec![tensor("Y", &dims)],
        vec![],
 
    
fn test_buffer_readability() {
    let _ = env_logger::builder().is_test(true).try_init();
    let n: usize = 16;
    let mut input_data = HashMap::new();

    let data = vec![0.0f32; n];
    let shape = vec![n as i64];
    input_data.insert("X".to_string(), data.as_slice().into());

    // Model: X -> Cos -> Y -> Flatten -> Z -> Flatten -> W
    let model = model(graph(
        vec![tensor("X", &shape)],
        vec![tensor("W", &shape)],
        vec![tensor("Y", &shape), tensor("Z", &shape)],
        vec![],
        vec![
            node(vec!["X"], vec!["Y"], "cos", "Cos", vec![]),
            node(vec!["Y"], vec!["Z"], "rs", "Reshape", vec![]),
            node(vec!["Z"], vec!["W"], "rs", "Reshape", vec![]),
        ],
    ));

    let session =
        pollster::block_on(wonnx::Session::from_model(model)).expect("Session did not create");

    let result = pollster::block_on(session.run(&input_data)).unwrap();
    assert_eq!(result["W"], OutputTensor::F32(vec![1.0; 16]));
}