TST: Add smoke test for plot_traj3d

soft-matter · Jun 7, 2024 · 10674d9 · 10674d9
1 parent 2d7c229
commit 10674d9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/trackpy/tests/test_plots.py b/trackpy/tests/test_plots.py
@@ -57,6 +57,12 @@ def test_ptraj_t_column(self):
         df.columns = cols
         plots.plot_traj(df, t_column='arbitrary name')
 
+    def test_ptraj3d(self):
+        sparse3d = self.sparse.copy()
+        sparse3d['z'] = 0
+        # smoke test
+        plots.plot_traj3d(sparse3d)
+
     def test_annotate(self):
         suppress_plotting()
         f = DataFrame({'x': [0, 1], 'y': [0, 1], 'frame': [0, 0],