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

Francisco Massa's avatar
Francisco Massa committed
3
import test_io
limm's avatar
limm committed
4
from torchvision import set_video_backend  # noqa: 401
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


limm's avatar
limm committed
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)