r3498 - /1.3/test_suite/unit_tests/generic_fns/test_spin.py -- November 11, 2007 - 21:26

Author: bugman
Date: Sun Nov 11 21:26:47 2007
New Revision: 3498

URL: http://svn.gna.org/viewcvs/relax?rev=3498&view=rev
Log:
Added 4 unit tests for the generic_fns.spin.delete() function.


Modified:
    1.3/test_suite/unit_tests/generic_fns/test_spin.py

Modified: 1.3/test_suite/unit_tests/generic_fns/test_spin.py
URL: 
http://svn.gna.org/viewcvs/relax/1.3/test_suite/unit_tests/generic_fns/test_spin.py?rev=3498&r1=3497&r2=3498&view=diff
==============================================================================
--- 1.3/test_suite/unit_tests/generic_fns/test_spin.py (original)
+++ 1.3/test_suite/unit_tests/generic_fns/test_spin.py Sun Nov 11 21:26:47 
2007
@@ -71,6 +71,12 @@
         cdp.mol[0].res[0].spin[0].num = 111
         cdp.mol[0].res[0].spin[0].name = 'C8'
         cdp.mol[0].res[0].spin[0].x = 1
+
+        # Add some more spins.
+        cdp.mol[0].res[0].spin.add_item('C19', 6)
+        cdp.mol[0].res[0].spin.add_item('C21', 7)
+        cdp.mol[0].res[0].spin.add_item('C24', 8)
+        cdp.mol[0].res[0].spin.add_item('C26', 9)
 
         # Create a second residue.
         cdp.mol[0].res.add_item('Arg', 2)
@@ -276,3 +282,75 @@
         self.assertRaises(RelaxError, spin.create, 1, 'P3')
 
 
+    def test_delete_name(self):
+        """Test spin deletion using spin name identifiers.
+
+        The function used is generic_fns.spin.delete().
+        """
+
+        # Set up the data.
+        self.setup_data()
+
+        # Delete the first spin.
+        spin.delete(spin_id='@C8')
+
+        # Test that the first spin is now 6, C19.
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].num, 
6)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].name, 'C19')
+        self.assert_(not 
hasattr(relax_data_store['orig'].mol[0].res[0].spin[0], 'x'))
+
+
+    def test_delete_num(self):
+        """Test spin deletion using spin number identifiers.
+
+        The function used is generic_fns.spin.delete().
+        """
+
+        # Set up the data.
+        self.setup_data()
+
+        # Delete the first spin.
+        spin.delete(spin_id=':111')
+
+        # Test that the first spin is now 6, C19.
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].num, 
6)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].name, 'C19')
+        self.assert_(not 
hasattr(relax_data_store['orig'].mol[0].res[0].spin[0], 'x'))
+
+
+    def test_delete_all(self):
+        """Test the deletion of all spins in one residue.
+
+        The function used is generic_fns.spin.delete().
+        """
+
+        # Set up the data.
+        self.setup_data()
+
+        # Delete all spins.
+        spin.delete(spin_id=':1-200')
+
+        # Test that the first spin defaults back to the empty spin.
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].num, 
None)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].name, None)
+
+
+    def test_delete_shift(self):
+        """Test the deletion of multiple spins.
+
+        The function used is generic_fns.spin.delete().
+        """
+
+        # Set up the data.
+        self.setup_data()
+
+        # Delete the first and third spins.
+        spin.delete(spin_id=':111,7')
+
+        # Test that the remaining spins.
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].num, 
6)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[0].name, 'C19')
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[1].num, 
8)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[1].name, 'C24')
+        self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[2].num, 
9)
+        
self.assertEqual(relax_data_store['orig'].mol[0].res[0].spin[2].name, 'C26')