test saved np.array can be loaded properly.

vanvalenlab · Jan 28, 2021 · c73cdeb · c73cdeb
1 parent 363f121
commit c73cdeb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/deepcell_tracking/utils_test.py b/deepcell_tracking/utils_test.py
@@ -125,6 +125,12 @@ def test_save_trks(self):
             utils.save_trks(filename, lineage, X, y)
             assert os.path.isfile(filename)
 
+            # test saved tracks can be loaded
+            loaded = utils.load_trks(filename)
+            assert loaded['lineages'] == lineage
+            np.testing.assert_array_equal(X, loaded['X'])
+            np.testing.assert_array_equal(y, loaded['y'])
+
         finally:
             try:
                 shutil.rmtree(tempdir)  # delete directory