def test_unflattenRanks_empty(self): t = Tensor(rank_ids=["X", "Y", "Z"]) t2 = t.flattenRanks() t3 = t2.unflattenRanks() t3.setRankIds(["X", "Y", "Z"]) self.assertEqual(t, t3)
def test_mutable_after_flatten(self): t = Tensor.fromYAMLfile("./data/test_tensor-1.yaml") t2 = t.flattenRanks() self.assertFalse(t2.isMutable()) t3 = Tensor(rank_ids=["X", "Y", "Z"]) t4 = t3.flattenRanks() self.assertTrue(t4.isMutable())