def testRankTwoIntTensor(self):
    x = np.array([[1, 2, 3], [4, 5, 6]])
    expected_ans = np.array([[[[1, 0, 0], [0, 0, 0]], [[0, 2, 0], [0, 0, 0]],
                              [[0, 0, 3], [0, 0, 0]]],
                             [[[0, 0, 0], [4, 0, 0]], [[0, 0, 0], [0, 5, 0]],