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

1from sleapyfaces.io import DAQData, SLEAPanalysis, BehMetadata, VideoMetadata 

2 

3 

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 ] 

28 

29 

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 

47 

48 

49def test_beh_initialization(): 

50 beh = BehMetadata( 

51 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/BehMetadata.json" 

52 ) 

53 

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 

64 

65 

66def test_video_initialization(): 

67 video = VideoMetadata( 

68 "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/video.mp4" 

69 ) 

70 

71 assert ( 

72 video.path 

73 == "/Users/annieehler/Projects/Jupyter_Notebooks/SLEAPyFaces/tests/data/20211105/video.mp4" 

74 ) 

75 

76 assert type(video.cache) is dict 

77 assert len(video.cache.keys()) > 3 

78 assert eval(video.cache["avg_frame_rate"]) == video.fps