Add a test

pytorch · Nov 22, 2020 · ef3a61f · ef3a61f
1 parent 47db4f9
commit ef3a61f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/test_datasets_video_utils.py b/test/test_datasets_video_utils.py
@@ -119,6 +119,16 @@ def test_compute_clips_for_video(self):
         self.assertTrue(clips.equal(idxs))
         self.assertTrue(idxs.flatten().equal(resampled_idxs))
 
+        # case 3: frames aren't enough for a clip
+        num_frames = 32
+        orig_fps = 30
+        new_fps = 13
+        with self.assertWarns(UserWarning):
+            clips, idxs = VideoClips.compute_clips_for_video(video_pts, num_frames, num_frames,
+                                                             orig_fps, new_fps)
+        self.assertEqual(len(clips), 0)
+        self.assertEqual(len(idxs), 0)
+
 
 if __name__ == '__main__':
     unittest.main()