Files
unshackle/tests/tracks/test_webvtt_timestamp_map.py
T

23 lines
935 B
Python

from unshackle.core.tracks.subtitle import Subtitle
def test_strips_timestamp_map_keeps_cues():
text = "WEBVTT\nX-TIMESTAMP-MAP=MPEGTS:900000,LOCAL:00:00:00.000\n\n00:00:22.690 --> 00:00:24.710\nHi\n"
out = Subtitle.strip_webvtt_timestamp_map(text)
assert "X-TIMESTAMP-MAP" not in out
assert "00:00:22.690 --> 00:00:24.710" in out
assert out.startswith("WEBVTT\n")
def test_strips_lowercase_timestamp_map():
# SubtitleEdit honours the header case-insensitively, so the strip must too
text = "WEBVTT\nx-timestamp-map=MPEGTS:900000,LOCAL:00:00:00.000\n\n00:00:22.690 --> 00:00:24.710\nHi\n"
out = Subtitle.strip_webvtt_timestamp_map(text)
assert "x-timestamp-map" not in out.lower()
assert "00:00:22.690 --> 00:00:24.710" in out
def test_no_timestamp_map_is_untouched():
text = "WEBVTT\n\n00:00:01.000 --> 00:00:02.000\nHi\n"
assert Subtitle.strip_webvtt_timestamp_map(text) == text