r9393 - /1.3/test_suite/unit_tests/_prompt/test_pymol.py -- August 26, 2009 - 11:26

Author: bugman
Date: Wed Aug 26 11:26:41 2009
New Revision: 9393

URL: http://svn.gna.org/viewcvs/relax?rev=9393&view=rev
Log:
Wrote 5 arg unit tests for the pymol.macro_exec() user function.


Added:
    1.3/test_suite/unit_tests/_prompt/test_pymol.py

Added: 1.3/test_suite/unit_tests/_prompt/test_pymol.py
URL: 
http://svn.gna.org/viewcvs/relax/1.3/test_suite/unit_tests/_prompt/test_pymol.py?rev=9393&view=auto
==============================================================================
--- 1.3/test_suite/unit_tests/_prompt/test_pymol.py (added)
+++ 1.3/test_suite/unit_tests/_prompt/test_pymol.py Wed Aug 26 11:26:41 2009
@@ -1,0 +1,104 @@
+###############################################################################
+#                                                                            
 #
+# Copyright (C) 2009 Edward d'Auvergne                                       
 #
+#                                                                            
 #
+# This file is part of the program relax.                                    
 #
+#                                                                            
 #
+# relax is free software; you can redistribute it and/or modify              
 #
+# it under the terms of the GNU General Public License as published by       
 #
+# the Free Software Foundation; either version 2 of the License, or          
 #
+# (at your option) any later version.                                        
 #
+#                                                                            
 #
+# relax 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 General Public License for more details.                               
 #
+#                                                                            
 #
+# You should have received a copy of the GNU General Public License          
 #
+# along with relax; if not, write to the Free Software                       
 #
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA  
 #
+#                                                                            
 #
+###############################################################################
+
+# Python module imports.
+from unittest import TestCase
+
+# relax module imports.
+from prompt.pymol import Pymol
+from relax_errors import RelaxNoneStrError, RelaxNoneStrListNumError, 
RelaxStrError
+
+# Unit test imports.
+from data_types import DATA_TYPES
+import fake_relax
+
+
+class Test_pymol(TestCase):
+    """Unit tests for the functions of the 'prompt.pymol' module."""
+
+    # Instantiate the user function class.
+    pymol_fns = Pymol(fake_relax.fake_instance())
+
+
+    def test_macro_exec_argfail_data_type(self):
+        """The data_type arg test of the pymol.macro_exec() user function."""
+
+        # Loop over the data types.
+        for data in DATA_TYPES:
+            # Catch the str arguments, and skip them.
+            if data[0] == 'str':
+                continue
+
+            # The argument test.
+            self.assertRaises(RelaxStrError, self.pymol_fns.macro_exec, 
data_type=data[1])
+
+
+    def test_macro_exec_argfail_style(self):
+        """The style arg test of the pymol.macro_exec() user function."""
+
+        # Loop over the data types.
+        for data in DATA_TYPES:
+            # Catch the str arguments, and skip them.
+            if data[0] == 'str':
+                continue
+
+            # The argument test.
+            self.assertRaises(RelaxStrError, self.pymol_fns.macro_exec, 
data_type='a', style=data[1])
+
+
+    def test_macro_exec_argfail_colour_start(self):
+        """The colour_start arg test of the pymol.macro_exec() user 
function."""
+
+        # Loop over the data types.
+        for data in DATA_TYPES:
+            # Catch the None, str, and num list arguments, and skip them.
+            if data[0] == 'None' or data[0] == 'str' or ((data[0] == 'int 
list' or data[0] == 'float list' or data[0] == 'number list') and 
len(data[1]) == 3):
+                continue
+
+            # The argument test.
+            self.assertRaises(RelaxNoneStrListNumError, 
self.pymol_fns.macro_exec, data_type='a', style='x', colour_start=data[1])
+
+
+    def test_macro_exec_argfail_colour_end(self):
+        """The colour_end arg test of the pymol.macro_exec() user 
function."""
+
+        # Loop over the data types.
+        for data in DATA_TYPES:
+            # Catch the None, str, and num list arguments, and skip them.
+            if data[0] == 'None' or data[0] == 'str' or ((data[0] == 'int 
list' or data[0] == 'float list' or data[0] == 'number list') and 
len(data[1]) == 3):
+                continue
+
+            # The argument test.
+            self.assertRaises(RelaxNoneStrListNumError, 
self.pymol_fns.macro_exec, data_type='a', style='x', colour_end=data[1])
+
+
+    def test_macro_exec_argfail_colour_list(self):
+        """The colour_list arg test of the pymol.macro_exec() user 
function."""
+
+        # Loop over the data types.
+        for data in DATA_TYPES:
+            # Catch the None and str arguments, and skip them.
+            if data[0] == 'None' or data[0] == 'str':
+                continue
+
+            # The argument test.
+            self.assertRaises(RelaxNoneStrError, self.pymol_fns.macro_exec, 
data_type='a', style='x', colour_list=data[1])