Upgrade the UnitTests to support unit test rules for different types, e.g. FullTensor and SparseTensor