refactor Op's unittest.
refactor Op's unittest.

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