fix save load test

jamesmkrieger · Dec 17, 2024 · be0883a · be0883a
1 parent 197fcf9
commit be0883a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/prody/tests/proteins/test_insty.py b/prody/tests/proteins/test_insty.py
@@ -53,7 +53,8 @@ def testAllInteractionsSave(self):
         """Test for saving and loading all types of interactions."""
         if prody.PY3K:
             self.INTERACTIONS_ALL = InteractionsTrajectory()
-            self.data_all = self.INTERACTIONS_ALL.calcProteinInteractionsTrajectory(self.ATOMS)
+            self.data_all = np.array(self.INTERACTIONS_ALL.calcProteinInteractionsTrajectory(self.ATOMS,
+                                                                                             stop_frame=13))
 
             np.save('test_2k39_all.npy', np.array(self.data_all, dtype=object), allow_pickle=True)