Added auto_save_result wrapper to save results from test cases to che…

…ck the files in Maya so I don't have to restart Maya every time.

tests/acceptance/test_biped.py

@@ -1,7 +1,7 @@
 import anvil.config as cfg
 import anvil.node_types as nt
 from anvil.rig_templates import Biped
-from tests.base_test import TestBase, sanitize
+from tests.base_test import TestBase, sanitize, auto_save_result
 
 
 class TestBaseTemplateRigs(TestBase):
@@ -11,6 +11,7 @@ class TestBaseTemplateRigs(TestBase):
     CLASS = Biped
 
     @classmethod
+    @auto_save_result
     def from_template_file(cls, template_file, **kwargs):
         cls.import_template_files(template_file)
 

tests/acceptance/test_hand.py

@@ -2,7 +2,7 @@
 import anvil.node_types as nt
 from anvil.utils.scene import print_scene_tree
 from anvil.sub_rig_templates import Hand
-from tests.base_test import TestBase, clean_up_scene
+from tests.base_test import TestBase, clean_up_scene, auto_save_result
 import string
 from pprint import pprint
 
@@ -14,6 +14,7 @@ class TestHandBase(TestBase):
     TEMPLATE_CLASS = Hand
 
     @classmethod
+    @auto_save_result
     def from_template_file(cls, template_file, finger_start_joints=None, **kwargs):
         cls.import_template_files(template_file)
 

tests/base_test.py

@@ -10,6 +10,7 @@
 from contextlib import contextmanager
 import nomenclate
 import anvil
+import anvil.runtime as rt
 from anvil.log import obtain_logger
 
 NOMENCLATE = nomenclate.Nom()
@@ -102,3 +103,18 @@ def wrapped(self, *args, **kwargs):
         return func_return
 
     return wrapped
+
+def auto_save_result(func):
+    @wraps(func)
+    def wrapped(self, *args, **kwargs):
+        save_file = kwargs.pop('save_file', False)
+        save_path = kwargs.pop('save_path', False)
+        func_return = func(self, *args, **kwargs)
+        if save_file:
+            rt.dcc.scene.fileop(rename=os.path.join([f for f in [save_path, save_file, 'mb'] if f]),
+                                save=True,
+                                type='mayaBinary')
+        return func_return
+
+    return wrapped
+