update unittests
update unittests

tensorflow/rust,adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,tensorflow/rust,adamcrume/tensorflow-rust,tensorflow/rust,tensorflow/rust