Fix bug in GRO reader

gmso/formats/gro.py

@@ -57,31 +57,30 @@
         coords = u.nm * np.zeros(shape=(n_atoms, 3))
         for row, _ in enumerate(coords):
             line = gro_file.readline()
-            content = line.split()
             if not line:
                 msg = (
                     "Incorrect number of lines in .gro file. Based on the "
                     "number in the second line of the file, {} rows of"
                     "atoms were expected, but at least one fewer was found."
                 )
                 raise ValueError(msg.format(n_atoms))
+            res_id = int(line[:5].strip())
+            res_name = line[5:10].strip()
+            atom_name = line[10:15].strip()
+            atom_id = line[15:20].strip()
 
-            res = content[0]
-            atom_name = content[1]
-            atom_id = content[2]
+            positions = line[20:].split()
             coords[row] = u.nm * np.array(
                 [
-                    float(content[3]),
-                    float(content[4]),
-                    float(content[5]),
+                    float(positions[0]),
+                    float(positions[1]),
+                    float(positions[2]),
                 ]
             )
             site = Atom(name=atom_name, position=coords[row])
 
-            r = re.compile("([0-9]+)([a-zA-Z]+)")
-            m = r.match(res)
-            site.molecule = (m.group(2), int(m.group(1)))
-            site.residue = (m.group(2), int(m.group(1)))
+            site.molecule = (res_name, res_id)
+            site.residue = (res_name, res_id)
             top.add_site(site, update_types=False)
         top.update_topology()
 

gmso/tests/test_gro.py

@@ -39,7 +39,7 @@ def test_read_gro(self):
         )
 
     def test_wrong_n_atoms(self):
-        with pytest.raises(IndexError):
+        with pytest.raises(ValueError):
             Topology.load(get_fn("too_few_atoms.gro"))
         with pytest.raises(ValueError):
             Topology.load(get_fn("too_many_atoms.gro"))