Fix IndexError for hdf5 file import for single instance analysis files

sleap/io/format/sleap_analysis.py

@@ -78,7 +78,7 @@ def read(
         # shape: frames * nodes * 2 * tracks
         frame_count, node_count, _, track_count = tracks_matrix.shape
 
-        if "track_names" in f:
+        if "track_names" in f and len(f["track_names"]):
             track_names_list = f["track_names"][:].T
             tracks = [Track(0, track_name.decode()) for track_name in track_names_list]
         else:

tests/fixtures/videos.py

@@ -1,12 +1,21 @@
 import pytest
 
 from sleap.io.video import Video
+from sleap.io.format.filehandle import FileHandle
 
 TEST_H5_FILE = "tests/data/hdf5_format_v1/training.scale=0.50,sigma=10.h5"
 TEST_H5_DSET = "/box"
 TEST_H5_CONFMAPS = "/confmaps"
 TEST_H5_AFFINITY = "/pafs"
 TEST_H5_INPUT_FORMAT = "channels_first"
+TEST_SMALL_ROBOT3_FRAME_H5 = (
+    "tests/data/hdf5_format_v1/small_robot.000_small_robot_3_frame.analysis.h5"
+)
+
+
+@pytest.fixture
+def small_robot_3_frame_hdf5():
+    return FileHandle(filename=TEST_SMALL_ROBOT3_FRAME_H5)
 
 
 @pytest.fixture

tests/io/test_formats.py

@@ -19,6 +19,19 @@
 from sleap.gui.app import MainWindow
 from sleap.gui.state import GuiState
 from sleap.info.write_tracking_h5 import get_nodes_as_np_strings
+from sleap.io.format.sleap_analysis import SleapAnalysisAdaptor
+
+
+def test_sleap_analysis_read(small_robot_3_frame_vid, small_robot_3_frame_hdf5):
+
+    # Single instance hdf5 analysis file test
+    read_labels = SleapAnalysisAdaptor.read(
+        file=small_robot_3_frame_hdf5, video=small_robot_3_frame_vid
+    )
+
+    assert len(read_labels.videos) == 1
+    assert len(read_labels.tracks) == 1
+    assert len(read_labels.skeletons) == 1
 
 
 def test_text_adaptor(tmpdir):