fixed tests

amepproject · Oct 21, 2024 · 3fd5720 · 3fd5720
1 parent 3212ed8
commit 3fd5720
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/test/test_trajectory.py b/test/test_trajectory.py
@@ -36,6 +36,7 @@
 FIELDDIR = DATADIR/'continuum'
 LAMMPSDIR = DATADIR/'lammps'
 HOOMDDIR = DATADIR/'hoomd'
+HOOMDDIR = DATADIR/'gromacs'
 
 RNG = np.random.default_rng(1234)
 
@@ -223,4 +224,17 @@ class TestHOOMDReader(unittest.TestCase):
     def test_trajfile(self):
         hoomd_traj = amep.load.traj(directory=HOOMDDIR,
                                     filename="simulated.gsd",
-                                    mode="gsd", start=.9)
+                                    mode="hoomd", start=.9)
+
+
+class TestGROMACSReader(unittest.TestCase):
+    """Testcase for HOOMD data.
+
+    TO BE IMPLEMENTED.
+    """
+
+    def test_trajfile(self):
+        gromacs_traj = amep.load.traj(directory=GROMACSDIR,
+                                    filename="traj.trr",
+                                    topology="top.tpr",
+                                    mode="gromacs", start=.9)