Make assertTensorEq work for all Tensor types. More tests.