def testRankOneIntTensor(self):
    x = np.array([1, 2, 3])
    expected_ans = np.array([[1, 0, 0], [0, 2, 0], [0, 0, 3]])
    self.diagOp(x, np.int32, expected_ans)
    self.diagOp(x, np.int64, expected_ans)