def test_shape_empty(self): """Test shape of empty tensor""" t1 = Tensor(rank_ids=["M", "K"]) self.assertEqual(t1.getRankIds(), ["M", "K"]) self.assertEqual(t1.getShape(), [0, 0]) t2 = Tensor(rank_ids=["M", "K"], shape=[10, 20]) self.assertEqual(t2.getRankIds(), ["M", "K"]) self.assertEqual(t2.getShape(), [10, 20])
def test_swizzleRanks_empty(self): """ Test swizzleRanks() on an empty tensor """ Z_MNOP = Tensor(rank_ids=["M", "N", "O", "P"]) Z_PNMO = Z_MNOP.swizzleRanks(rank_ids=["P", "N", "M", "O"]) self.assertEqual(Z_MNOP.getRankIds(), ["M", "N", "O", "P"]) self.assertEqual(Z_PNMO.getRankIds(), ["P", "N", "M", "O"])
def test_constructor_empty(self): """Test construction of empty tensor""" ranks = ["M", "K"] t = Tensor(rank_ids=ranks) self.assertEqual(t.getRankIds(), ranks) self.assertEqual(t.getRoot().getRankIds(), ranks)
def test_shape_0D(self): """Test shpe of 0-D tensor""" t = Tensor(rank_ids=[]) p = t.getRoot() p += 1 self.assertEqual(t.getRankIds(), []) self.assertEqual(t.getShape(), [])
def test_constructor_shape(self): """Test construction of shape of tensor""" ranks = ["M", "K"] shape = [4, 8] t = Tensor(rank_ids=ranks, shape=shape) self.assertEqual(t.getRankIds(), ranks) self.assertEqual(t.getRoot().getRankIds(), ranks) self.assertEqual(t.getShape(), shape) self.assertEqual(t.getRoot().getShape(), shape)