Faster unit tests

This PR reduces the sizes of the Tensors and operations used in the unit tests to make faster. From 90-ish seconds to 6-ish seconds.