skip segfaulting tests for python 3.9

Signed-off-by: Eli Uriegas <[email protected]>

Author: seemethere

test/common_utils.py

@@ -18,6 +18,9 @@
 import numpy as np
 from PIL import Image
 
+IS_PY39 = sys.version_info.major == 3 and sys.version_info.minor == 9
+PY39_SEGFAULT_SKIP_MSG = "Segmentation fault with Python 3.9, see https://github.com/pytorch/vision/issues/3367"
+PY39_SKIP = unittest.skipIf(IS_PY39, PY39_SEGFAULT_SKIP_MSG)
 
 @contextlib.contextmanager
 def get_tmp_dir(src=None, **kwargs):

test/test_video.py

@@ -4,6 +4,7 @@
 import tempfile
 import unittest
 import random
+import sys
 
 import itertools
 
@@ -13,6 +14,7 @@
 import torch
 import torchvision
 from torchvision.io import _HAS_VIDEO_OPT, VideoReader
+from common_utils import PY39_SKIP
 
 try:
     import av
@@ -278,6 +280,7 @@ def _template_read_video(video_object, s=0, e=None):
 
 @unittest.skipIf(_HAS_VIDEO_OPT is False, "Didn't compile with ffmpeg")
 class TestVideo(unittest.TestCase):
+    @PY39_SKIP
     @unittest.skipIf(av is None, "PyAV unavailable")
     def test_read_video_tensor(self):
         """
@@ -362,6 +365,7 @@ def test_metadata(self):
                 config.duration, reader_md["video"]["duration"][0], delta=0.5
             )
 
+    @PY39_SKIP
     @unittest.skipIf(av is None, "PyAV unavailable")
     def test_video_reading_fn(self):
         """

test/test_video_reader.py

@@ -10,6 +10,7 @@
 import torchvision.io as io
 from numpy.random import randint
 from torchvision.io import _HAS_VIDEO_OPT
+from common_utils import PY39_SKIP
 
 
 try:
@@ -426,6 +427,7 @@ def test_stress_test_read_video_from_file(self):
                     audio_timebase_den,
                 )
 
+    @PY39_SKIP
     def test_read_video_from_file(self):
         """
         Test the case when decoder starts with a video file to decode frames.
@@ -471,6 +473,7 @@ def test_read_video_from_file(self):
             # compare decoding results
             self.compare_decoding_result(tv_result, pyav_result, config)
 
+    @PY39_SKIP
     def test_read_video_from_file_read_single_stream_only(self):
         """
         Test the case when decoder starts with a video file to decode frames, and
@@ -779,6 +782,7 @@ def test_read_video_from_file_rescale_width_and_height(self):
             self.assertEqual(tv_result[0].size(1), height)
             self.assertEqual(tv_result[0].size(2), width)
 
+    @PY39_SKIP
     def test_read_video_from_file_audio_resampling(self):
         """
         Test the case when decoder starts with a video file to decode frames, and
@@ -838,6 +842,7 @@ def test_read_video_from_file_audio_resampling(self):
                         delta=0.1 * asample_rate.item(),
                     )
 
+    @PY39_SKIP
     def test_compare_read_video_from_memory_and_file(self):
         """
         Test the case when video is already in memory, and decoder reads data in memory
@@ -904,6 +909,7 @@ def test_compare_read_video_from_memory_and_file(self):
             # finally, compare results decoded from memory and file
             self.compare_decoding_result(tv_result_memory, tv_result_file)
 
+    @PY39_SKIP
     def test_read_video_from_memory(self):
         """
         Test the case when video is already in memory, and decoder reads data in memory
@@ -948,6 +954,7 @@ def test_read_video_from_memory(self):
             self.check_separate_decoding_result(tv_result, config)
             self.compare_decoding_result(tv_result, pyav_result, config)
 
+    @PY39_SKIP
     def test_read_video_from_memory_get_pts_only(self):
         """
         Test the case when video is already in memory, and decoder reads data in memory.
@@ -1017,6 +1024,7 @@ def test_read_video_from_memory_get_pts_only(self):
             self.assertEqual(tv_result_pts_only[5].numel(), 0)
             self.compare_decoding_result(tv_result, tv_result_pts_only)
 
+    @PY39_SKIP
     def test_read_video_in_range_from_memory(self):
         """
         Test the case when video is already in memory, and decoder reads data in memory.
@@ -1192,6 +1200,7 @@ def test_probe_video_from_memory_script(self):
             probe_result = scripted_fun(video_tensor)
             self.check_meta_result(probe_result, config)
 
+    @PY39_SKIP
     def test_read_video_from_memory_scripted(self):
         """
         Test the case when video is already in memory, and decoder reads data in memory