r4239 - /branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py -- December 21, 2007 - 15:58

Author: bugman
Date: Fri Dec 21 15:58:56 2007
New Revision: 4239

URL: http://svn.gna.org/viewcvs/relax?rev=4239&view=rev
Log:
Wrote a unit test for the align_tensor.matrix_angles() functions.


Modified:
    branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py

Modified: 
branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py
URL: 
http://svn.gna.org/viewcvs/relax/branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py?rev=4239&r1=4238&r2=4239&view=diff
==============================================================================
--- branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py 
(original)
+++ branches/N_state_model/test_suite/unit_tests/align_tensor_testing_base.py 
Fri Dec 21 15:58:56 2007
@@ -19,6 +19,9 @@
 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA  
 #
 #                                                                            
 #
 
###############################################################################
+
+# Python module imports.
+from math import pi
 
 # relax module imports.
 from data import Data as relax_data_store
@@ -222,6 +225,55 @@
         self.assertEqual(relax_data_store['orig'].align_tensor['Pf1'].Sxy, 
7.65639)
         self.assertEqual(relax_data_store['orig'].align_tensor['Pf1'].Sxz, 
-1.89157)
         self.assertEqual(relax_data_store['orig'].align_tensor['Pf1'].Syz, 
19.2561)
+
+
+    def test_matrix_angles_identity(self):
+        """Test the matrix angles for a 5x5 identity matrix.
+
+        The functions tested are both 
generic_fns.align_tensor.matrix_angles() and
+        prompt.align_tensor.matrix_angles().
+        """
+
+        # Initialise the 5 tensors.
+        self.align_tensor_fns.init(tensor='1', params=(1, 0, 0, 0, 0))
+        self.align_tensor_fns.init(tensor='2', params=(0, 1, 0, 0, 0))
+        self.align_tensor_fns.init(tensor='3', params=(0, 0, 1, 0, 0))
+        self.align_tensor_fns.init(tensor='4', params=(0, 0, 0, 1, 0))
+        self.align_tensor_fns.init(tensor='5', params=(0, 0, 0, 0, 1))
+
+        # Matrix angles.
+        self.align_tensor_fns.matrix_angles()
+
+        # Test the angles.
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[0,0], 
0.0)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[0,1], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[0,2], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[0,3], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[0,4], 
pi/2)
+
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[1,0], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[1,1], 
0.0)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[1,2], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[1,3], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[1,4], 
pi/2)
+
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[2,0], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[2,1], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[2,2], 
0.0)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[2,3], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[2,4], 
pi/2)
+
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[3,0], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[3,1], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[3,2], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[3,3], 
0.0)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[3,4], 
pi/2)
+
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[4,0], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[4,1], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[4,2], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[4,3], 
pi/2)
+        self.assertEqual(relax_data_store['orig'].align_tensor.angles[4,4], 
0.0)
 
 
     def test_svd_identity(self):