Wrote tests for small tensor class.