| """Regression: never add an extra frozen forecast tail to new encodes.""" |
| import json |
| import tempfile |
| import unittest |
| from pathlib import Path |
| from unittest.mock import patch |
|
|
| from build_fung_wong_video import encode_video |
|
|
|
|
| class VideoPlaybackTest(unittest.TestCase): |
| def test_all_twenty_states_without_padding(self): |
| with tempfile.TemporaryDirectory(dir='/Volumes/D/typhoon_predict/output') as name: |
| work = Path(name) |
| for i in range(20): |
| (work/f'frame_{i:03d}.png').touch() |
| video = work/'video.mp4' |
| video.touch() |
| probe = {'streams': [{'codec_name': 'h264', 'width': 1600, |
| 'height': 1000, 'nb_frames': '600'}], 'format': {'duration': '20'}} |
| with patch('build_fung_wong_video.subprocess.run') as run, patch( |
| 'build_fung_wong_video.subprocess.check_output', return_value=json.dumps(probe).encode()): |
| result = encode_video(work, video) |
| command = run.call_args.args[0] |
| self.assertNotIn('tpad', ' '.join(command)) |
| self.assertEqual(command[command.index('-frames:v')+1], '600') |
| self.assertEqual(result['extra_final_hold_seconds'], 0) |
| self.assertEqual(result['duration_seconds'], 20) |
|
|
| def test_missing_state_is_not_repeated(self): |
| with tempfile.TemporaryDirectory(dir='/Volumes/D/typhoon_predict/output') as name: |
| with self.assertRaisesRegex(ValueError, 'Missing or unexpected'): |
| encode_video(Path(name), Path(name)/'video.mp4') |
|
|
|
|
| if __name__ == '__main__': |
| unittest.main() |
|
|
