Coverage for tests/test_io.py: 100%
32 statements
« prev ^ index » next coverage.py v7.0.2, created at 2023-01-03 12:07 -0800
« prev ^ index » next coverage.py v7.0.2, created at 2023-01-03 12:07 -0800
1from sleapyfaces.io import DAQData, SLEAPanalysis, BehMetadata, VideoMetadata
4def test_daq_initialization():
5 daq = DAQData(
6 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/DAQOutput.csv"
7 )
8 assert (
9 daq.path
10 == "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/DAQOutput.csv"
11 )
12 assert daq.cache.shape[1] == 13
13 assert daq.cache.shape[0] > 13
14 assert daq.columns == [
15 "Shutter_on",
16 "Shutter_off",
17 "LED590_on",
18 "LED590_off",
19 "Lick_on",
20 "Lick_off",
21 "Speaker_on",
22 "Speaker_off",
23 "Airpuff_on",
24 "Airpuff_off",
25 "Sucrose_on",
26 "Sucrose_off",
27 ]
30def test_sleap_initialization():
31 sleap = SLEAPanalysis(
32 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/SLEAP.h5"
33 )
34 assert (
35 sleap.path
36 == "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/SLEAP.h5"
37 )
38 names = [0] * len(sleap.data["node_names"]) * 2
39 for name, i in zip(
40 sleap.data["node_names"], range(0, (len(sleap.data["node_names"]) * 2), 2)
41 ):
42 names[i] = f"{name.replace(' ', '_')}_x"
43 names[i + 1] = f"{name.replace(' ', '_')}_y"
44 assert len(sleap.data.keys()) > 4
45 assert len(sleap.tracks.columns) == (len(sleap.data["node_names"]) * 2)
46 assert sleap.track_names == names
49def test_beh_initialization():
50 beh = BehMetadata(
51 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/BehMetadata.json"
52 )
54 assert beh.ITIArrayKey == "ITIArray"
55 assert beh.TrialArrayKey == "trialArray"
56 assert beh.MetaDataKey == "beh_metadata"
57 assert (
58 beh.path
59 == "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/BehMetadata.json"
60 )
61 assert beh.columns == ["trialArray", "ITIArray"]
62 assert beh.cache.shape[1] == 2
63 assert beh.cache.shape[0] > 2
66def test_video_initialization():
67 video = VideoMetadata(
68 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/video.mp4"
69 )
71 assert (
72 video.path
73 == "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/video.mp4"
74 )
76 assert type(video.cache) is dict
77 assert len(video.cache.keys()) > 3
78 assert eval(video.cache["avg_frame_rate"]) == video.fps