From 0974ac1544f7c346daefc924d586f75386ca45dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Rodrigues?= <j.p.g.l.m.rodrigues@gmail.com>
Date: Thu, 29 Nov 2018 11:21:48 -0800
Subject: [PATCH] Fixed test case after bug on tool

---
 tests/test_pdb_tocif.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_pdb_tocif.py b/tests/test_pdb_tocif.py
index c5fec6d7..4ae148f1 100644
--- a/tests/test_pdb_tocif.py
+++ b/tests/test_pdb_tocif.py
@@ -64,7 +64,7 @@ def test_single_model(self):
 
         # Validate results
         self.assertEqual(self.retcode, 0)
-        self.assertEqual(len(self.stdout), 211)
+        self.assertEqual(len(self.stdout), 212)
         self.assertEqual(len(self.stderr), 0)
 
         # Check no of records
@@ -75,7 +75,7 @@ def test_single_model(self):
         self.assertEqual(n_HETATM, 9)
         self.assertEqual(n_coord, 185)
 
-    def test_valid(self):
+    def test_multi_model(self):
         """$ pdb_tocif data/ensemble_OK.pdb"""
 
         fpath = os.path.join(data_dir, 'ensemble_OK.pdb')
@@ -86,7 +86,7 @@ def test_valid(self):
 
         # Validate results
         self.assertEqual(self.retcode, 0)
-        self.assertEqual(len(self.stdout), 30)
+        self.assertEqual(len(self.stdout), 31)
         self.assertEqual(len(self.stderr), 0)
 
         # Check no of records
@@ -105,7 +105,7 @@ def test_valid(self):
         # Check number of fields
         atom_lines = [l for l in self.stdout if l.startswith(records)]
         n_fields = list(set(map(lambda x: len(x.split()), atom_lines)))
-        self.assertEqual(n_fields, [20])
+        self.assertEqual(n_fields, [21])
 
     def test_file_not_found(self):
         """$ pdb_tocif not_existing.pdb"""