| 78 | |
| 79 | |
| 80 | def test_opencv_video_format(codec, file_ext): |
| 81 | with tempfile.TemporaryDirectory(prefix="video_format_test") as dir: |
| 82 | filename = os.path.join(dir, "test_file" + file_ext) |
| 83 | writer = cv2.VideoWriter( |
| 84 | filename=filename, |
| 85 | fourcc=cv2.VideoWriter_fourcc(*codec), |
| 86 | fps=float(30), |
| 87 | frameSize=(10, 10), |
| 88 | isColor=True, |
| 89 | ) |
| 90 | [writer.write(np.zeros((10, 10, 3), np.uint8)) for _ in range(30)] |
| 91 | writer.release() |
| 92 | if os.path.isfile(filename): |
| 93 | return True |
| 94 | return False |
| 95 | |
| 96 | |
| 97 | if __name__ == "__main__": |