Add formatting fixtures, tracking_scores test

tests/fixtures/datasets.py

@@ -21,6 +21,8 @@
 TEST_SLP_MIN_LABELS_ROBOT = "tests/data/slp_hdf5/small_robot_minimal.slp"
 TEST_MIN_TRACKS_2NODE_LABELS = "tests/data/tracks/clip.2node.slp"
 TEST_MIN_TRACKS_13NODE_LABELS = "tests/data/tracks/clip.slp"
+TEST_HDF5_PREDICTIONS = "tests/data/hdf5_format_v1/centered_pair_predictions.h5"
+TEST_SLP_PREDICTIONS = "tests/data/hdf5_format_v1/centered_pair_predictions.slp"
 
 
 @pytest.fixture
@@ -213,3 +215,13 @@ def multi_skel_vid_labels(hdf5_vid, small_robot_mp4_vid, skeleton, stickman):
     labels = Labels(labels)
 
     return labels
+
+
+@pytest.fixture
+def centered_pair_predictions_hdf5_path():
+    return TEST_HDF5_PREDICTIONS
+
+
+@pytest.fixture
+def centered_pair_predictions_slp_path():
+    return TEST_SLP_PREDICTIONS

tests/io/test_formats.py

@@ -82,8 +82,8 @@ def test_missing_file():
         disp.read("missing_file.txt")
 
 
-def test_hdf5_v1(tmpdir):
-    filename = "tests/data/hdf5_format_v1/centered_pair_predictions.h5"
+def test_hdf5_v1(tmpdir, centered_pair_predictions_hdf5_path):
+    filename = centered_pair_predictions_hdf5_path
     disp = dispatch.Dispatch.make_dispatcher(adaptor.SleapObjectType.labels)
 
     # Make sure reading works
@@ -99,8 +99,9 @@ def test_hdf5_v1(tmpdir):
     assert len(y.labeled_frames) == 1100
 
 
-def test_hdf5_v1_filehandle():
-    filename = "tests/data/hdf5_format_v1/centered_pair_predictions.h5"
+def test_hdf5_v1_filehandle(centered_pair_predictions_hdf5_path):
+
+    filename = centered_pair_predictions_hdf5_path
 
     labels = hdf5.LabelsV1Adaptor.read_headers(filehandle.FileHandle(filename))
 
@@ -142,11 +143,11 @@ def test_json_v1(tmpdir, centered_pair_labels):
     assert len(y.labeled_frames) == len(centered_pair_labels.labeled_frames)
 
 
-def test_matching_adaptor():
+def test_matching_adaptor(centered_pair_predictions_hdf5_path):
     from sleap.io.format import read
 
     read(
-        "tests/data/hdf5_format_v1/centered_pair_predictions.h5",
+        centered_pair_predictions_hdf5_path,
         for_object="labels",
         as_format="*",
     )
@@ -188,3 +189,27 @@ def test_madlc(test_data):
     assert_array_equal(labels[1][1].numpy(), [[17, 18], [np.nan, np.nan], [20, 21]])
     assert_array_equal(labels[2][0].numpy(), [[22, 23], [24, 25], [26, 27]])
     assert labels[2].frame_idx == 3
+
+
+def test_tracking_scores(tmpdir, centered_pair_predictions_slp_path):
+
+    # test reading
+    filename = centered_pair_predictions_slp_path
+
+    fh = filehandle.FileHandle(filename)
+
+    assert fh.format_id is not None
+
+    labels = hdf5.LabelsV1Adaptor.read(fh)
+
+    for instance in labels.instances():
+        assert hasattr(instance, "tracking_score")
+
+    # test writing
+    filename = os.path.join(tmpdir, "test.slp")
+    labels.save(filename)
+
+    labels = hdf5.LabelsV1Adaptor.read(filehandle.FileHandle(filename))
+
+    for instance in labels.instances():
+        assert hasattr(instance, "tracking_score")