test_io_opt.py 308 Bytes
Newer Older
Francisco Massa's avatar
Francisco Massa committed
1
import unittest
2

Francisco Massa's avatar
Francisco Massa committed
3
import test_io
4
from torchvision import set_video_backend
Francisco Massa's avatar
Francisco Massa committed
5
6


7
8
# Disabling the video backend switching temporarily
# set_video_backend('video_reader')
Francisco Massa's avatar
Francisco Massa committed
9
10


11
if __name__ == "__main__":
Francisco Massa's avatar
Francisco Massa committed
12
13
    suite = unittest.TestLoader().loadTestsFromModule(test_io)
    unittest.TextTestRunner(verbosity=1).run(suite)