From 4dbc84737c3d99fe3a015796eadfe67b72c416a6 Mon Sep 17 00:00:00 2001
From: Gerardo Tauriello <gerardo.tauriello@unibas.ch>
Date: Thu, 4 Jul 2019 19:09:03 +0200
Subject: [PATCH] Extend PDBize unit test.

---
 modules/io/tests/test_io_mmcif.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/modules/io/tests/test_io_mmcif.py b/modules/io/tests/test_io_mmcif.py
index dfd12ee30..e0672fc39 100644
--- a/modules/io/tests/test_io_mmcif.py
+++ b/modules/io/tests/test_io_mmcif.py
@@ -167,14 +167,21 @@ class TestMMCifInfo(unittest.TestCase):
 
     self.assertEquals(''.join([c.name for c in pdb_ent.chains]),
                       'AB_-CDEFGH')
-                      
     self.assertEquals([c.residue_count for c in pdb_ent.chains],
-                      [415, 414, 64, 3816, 415,414,415,414,415,414])
+                      [415, 414, 64, 3816, 415, 414, 415, 414, 415, 414])
+    self.assertEquals([c.atom_count for c in pdb_ent.chains],
+                      [3231, 3223, 268, 3816, 3231, 3223, 3231, 3223, 3231, 3223])
+    self.assertEquals([c.GetBondCount() for c in pdb_ent.chains],
+                      [3311, 3303, 204, 0, 3311, 3303, 3311, 3303, 3311, 3303])
 
     self.assertEquals(''.join([c.name for c in pdb_seqres_ent.chains]),
-                      'AB_-CDEFGH')
+                      ''.join([c.name for c in pdb_ent.chains]))
     self.assertEquals([c.residue_count for c in pdb_seqres_ent.chains],
-                      [415, 414, 64, 3816, 415,414,415,414,415,414])
+                      [c.residue_count for c in pdb_ent.chains])
+    self.assertEquals([c.atom_count for c in pdb_seqres_ent.chains],
+                      [c.atom_count for c in pdb_ent.chains])
+    self.assertEquals([c.GetBondCount() for c in pdb_seqres_ent.chains],
+                      [c.GetBondCount() for c in pdb_ent.chains])
 
   def test_mmcifinfo_biounit_pdbize_transformation(self):
     ent, seqres, info = io.LoadMMCIF("testfiles/mmcif/3hqv.cif.gz",
-- 
GitLab