fix unite chains test for default False

jamesmkrieger · Oct 16, 2023 · e6d15d7 · e6d15d7
1 parent 14ecff6
commit e6d15d7
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/prody/tests/proteins/test_ciffile.py b/prody/tests/proteins/test_ciffile.py
@@ -121,18 +121,23 @@ def testUniteChainsArgument(self):
 
         path = pathDatafile(self.biomols['file'])
         self.assertEqual(parseMMCIF(path, chain='A').numAtoms(), 
+                        self.biomols['chainA_atoms_alone'],
+                        'parseMMCIF failed to parse correct number of atoms '
+                        'when chain A is specified with unite_chain default (False)')
+        self.assertEqual(parseMMCIF(path, chain='A', unite_chains=True).numAtoms(), 
                         self.biomols['chainA_atoms_united'],
                         'parseMMCIF failed to parse correct number of atoms '
-                        'when chain A is specified with unite_chain default (True)')
-        self.assertEqual(parseMMCIF(path, chain='A', unite_chains=False).numAtoms(), 
+                        'when chain A is specified with unite_chain True')
+        self.assertEqual(parseMMCIF(path, chain='A', header=True)[0].numAtoms(), 
                         self.biomols['chainA_atoms_alone'],
                         'parseMMCIF failed to parse correct number of atoms '
-                        'when chain A is specified with unite_chain False')
-        self.assertEqual(parseMMCIF(path, chain='A', header=True)[0].numAtoms(), 
+                        'when chain A is specified with unite_chain default (False) '
+                        'with header True')
+        self.assertEqual(parseMMCIF(path, chain='A', header=True, unite_chains=True)[0].numAtoms(), 
                         self.biomols['chainA_atoms_united'],
                         'parseMMCIF failed to parse correct number of atoms '
-                        'when chain A is specified with unite_chain default (True) '
-                        'with header True')        
+                        'when chain A is specified with unite_chain True '
+                        'with header True')
 
     def testUniteChainsAndBiomolArguments(self):
         """Test outcome of valid and invalid *segment* arguments."""