Update

2023/scripts/animation_tools/studiolibrary/mutils/tests/test_anim.py

@@ -0,0 +1,125 @@
+# Copyright 2020 by Kurt Rathjen. All Rights Reserved.
+#
+# This library is free software: you can redistribute it and/or modify it 
+# under the terms of the GNU Lesser General Public License as published by 
+# the Free Software Foundation, either version 3 of the License, or 
+# (at your option) any later version. This library is distributed in the 
+# hope that it will be useful, but WITHOUT ANY WARRANTY; without even the 
+# implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
+# See the GNU Lesser General Public License for more details.
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library. If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+
+import mutils
+
+import test_base
+
+
+class TestAnim(test_base.TestBase):
+
+    def setUp(self):
+        """
+        """
+
+    def save(self, bakeConnected=False):
+        """
+        Test saving the animation to disc.
+        """
+        self.open()
+        anim = mutils.Animation.fromObjects(self.srcObjects)
+        anim.save(self.dstPath, bakeConnected=bakeConnected)
+
+    # def test_older_version(self):
+    #     """
+    #     Test animation parser for an older animation format
+    #     """
+    #     srcPath = self.dataPath("test_older_version.anim")
+    #     a = mutils.Animation.fromPath(srcPath)
+
+    def test_load_replace_completely(self):
+        """
+        Test loading the animation with replace completely option.
+        """
+        self.srcPath = self.dataPath("test_anim.ma")
+        self.dstPath = self.dataPath("test_load_replace_completely.anim")
+        self.save()
+
+        anim = mutils.Animation.fromPath(self.dstPath)
+        anim.load(self.dstObjects)
+
+        self.assertEqualAnimation()
+
+    def test_bake_connected(self):
+        """
+        Test saving animation with the option bake connected.
+        """
+        srcPath = self.dataPath("test_bake_connected.ma")
+        dstPath = self.dataPath("test_bake_connected.anim")
+
+        srcObjects = [
+            "srcSphere:group",
+            "srcSphere:lockedNode",
+            "srcSphere:offset",
+            "srcSphere:sphere"
+        ]
+
+        dstObjects = [
+            "dstSphere:group",
+            "dstSphere:lockedNode",
+            "dstSphere:offset",
+            "dstSphere:sphere"
+        ]
+
+        self.open(path=srcPath)
+
+        anim = mutils.Animation.fromObjects(srcObjects)
+        anim.save(dstPath, bakeConnected=True)
+
+        anim = mutils.Animation.fromPath(dstPath)
+        anim.load(dstObjects)
+
+        self.assertEqualAnimation()
+
+    def test_load_replace(self):
+        """
+        Test loading the animation with the option Replace.
+        """
+        self.srcPath = self.dataPath("test_anim.ma")
+        self.dstPath = self.dataPath("test_load_replace.anim")
+        self.save()
+
+        anim = mutils.Animation.fromPath(self.dstPath)
+        anim.load(self.dstObjects, option=mutils.PasteOption.Replace, startFrame=5)
+
+    def test_load_insert(self):
+        """
+        Test loading the animation with the option Insert.
+        """
+        self.srcPath = self.dataPath("test_anim.ma")
+        self.dstPath = self.dataPath("test_load_insert.anim")
+        self.save()
+
+        anim = mutils.Animation.fromPath(self.dstPath)
+        anim.load(self.dstObjects, option=mutils.PasteOption.Insert, startFrame=5)
+
+
+def testSuite():
+    """
+    Return the test suite for the test case.
+
+    :rtype: unittest.TestSuite
+    """
+    suite = unittest.TestSuite()
+    s = unittest.makeSuite(TestAnim, 'test')
+    suite.addTest(s)
+    return suite
+
+
+def run():
+    """
+    Call from within Maya to run all valid tests.
+    """
+    tests = unittest.TextTestRunner()
+    tests.run(testSuite())