126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
# 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())
|