r27063 - /trunk/test_suite/system_tests/structure.py -- December 10, 2014 - 18:53

Author: bugman
Date: Wed Dec 10 18:53:48 2014
New Revision: 27063

URL: http://svn.gna.org/viewcvs/relax?rev=27063&view=rev
Log:
Created the Structure.test_find_pivot system test.

This is to check the structure.find_pivot user function as this algorithm is 
currently not being
checked in the test suite.


Modified:
    trunk/test_suite/system_tests/structure.py

Modified: trunk/test_suite/system_tests/structure.py
URL: 
http://svn.gna.org/viewcvs/relax/trunk/test_suite/system_tests/structure.py?rev=27063&r1=27062&r2=27063&view=diff
==============================================================================
--- trunk/test_suite/system_tests/structure.py  (original)
+++ trunk/test_suite/system_tests/structure.py  Wed Dec 10 18:53:48 2014
@@ -3127,6 +3127,37 @@
                         
self.assertAlmostEqual(cdp.structure.displacements._rotation_axis[i][j][k], 
rot_axis[i][j][k])
 
 
+    def test_find_pivot(self):
+        """Test the structure.find_pivot user function.
+
+        This checks the default operation of the U{structure.find_pivot user 
function<http://www.nmr-relax.com/manual/structure_find_pivot.html>}.
+        """
+
+        # Path of the PDB file.
+        path = status.install_path + 
sep+'test_suite'+sep+'shared_data'+sep+'structures'
+
+        # Load the PDB three times as different models.
+        self.interpreter.structure.read_pdb('Ap4Aase_res1-12.pdb', dir=path, 
set_model_num=1)
+        self.interpreter.structure.read_pdb('Ap4Aase_res1-12.pdb', dir=path, 
set_model_num=2)
+        self.interpreter.structure.read_pdb('Ap4Aase_res1-12.pdb', dir=path, 
set_model_num=3)
+
+        # Rotate two of the models (the pivot will be the origin).
+        pivot = [1., 2., 3.]
+        R = zeros((3, 3), float64)
+        axis_angle_to_R(array([1, 0, 0], float64), 0.1, R)
+        self.interpreter.structure.rotate(R=R, model=2, origin=pivot)
+        axis_angle_to_R(array([0, 1, 0], float64), 0.2, R)
+        self.interpreter.structure.rotate(R=R, model=3, origin=pivot)
+
+        # Find the pivot.
+        self.interpreter.structure.find_pivot(init_pos=[0.95, 2.05, 3.02])
+
+        # Check the pivot.
+        self.assertAlmostEqual(cdp.structure.pivot[0], pivot[0], 3)
+        self.assertAlmostEqual(cdp.structure.pivot[1], pivot[1], 3)
+        self.assertAlmostEqual(cdp.structure.pivot[2], pivot[2], 3)
+
+
     def test_get_model(self):
         """Test the get_model() method of the internal structural object."""