fix: fix count of frames in trajectory

jag1g13 · Oct 22, 2020 · d02cd34 · d02cd34
1 parent e1186eb
commit d02cd34
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pycgtool/frame.py b/pycgtool/frame.py
@@ -136,10 +136,9 @@ def natoms(self) -> int:
         return self._topology.n_atoms
 
     @property
-    def numframes(self) -> int:
-        """Number of frames in the frame trajectory."""
-        # The MDTraj trajectory has the topology file as frame 0
-        return self._trajectory.n_frames - 1
+    def n_frames(self) -> int:
+        """Number of frames in the trajectory."""
+        return self._trajectory.n_frames
 
     def add_residue(self,
                     name,

diff --git a/test/test_frame.py b/test/test_frame.py
@@ -112,7 +112,7 @@ def test_frame_output_gro(self):
 
     def test_frame_read_xtc_numframes(self):
         frame = Frame('test/data/water.gro', 'test/data/water.xtc')
-        self.assertEqual(10, frame.numframes)
+        self.assertEqual(11, frame.n_frames)
 
     def test_frame_read_xtc(self):
         frame = Frame('test/data/water.gro', 'test/data/water.xtc')