Python Tensor.flattenRanksの例

コード例 #1

ファイル: test_tensor_transform.py プロジェクト: Fibertree-Project/fibertree

    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)

コード例 #2

ファイル: test_tensor.py プロジェクト: Fibertree-Project/fibertree

    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())