Test cell in movie must be in lineage.

vanvalenlab · Sep 7, 2021 · b11d907 · b11d907
1 parent 62f9d35
commit b11d907
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/deepcell_tracking/utils_test.py b/deepcell_tracking/utils_test.py
@@ -368,11 +368,8 @@ def test_is_valid_lineage(self):
         assert not utils.is_valid_lineage(movie, bad_lineage)
 
         # cell in movie but not in lineage is invalid
-        new_frame = get_annotated_image(num_labels=1, sequential=False)
-        bad_movie = np.concatenate([
-            movie,
-            np.expand_dims(new_frame, axis=0)
-        ], axis=0)
+        bad_movie = copy.deepcopy(movie)
+        bad_movie[0, 0, 0] = bad_label
         assert not utils.is_valid_lineage(bad_movie, lineage)
 
         # a daughter's frames should start immediatlely