def test_start_seconds(start_seconds): parser_zero = SrtParser(start_seconds=0) parser_zero.fit(BytesIO(fake_srt)) parser = SrtParser(start_seconds=start_seconds) parser.fit(BytesIO(fake_srt)) expected = [sub for sub in parser_zero.subs_ if sub.start >= timedelta(seconds=start_seconds)] assert expected == parser.subs_
def test_max_seconds(max_seconds): parser = SrtParser(max_subtitle_seconds=max_seconds) parser.fit(BytesIO(fake_srt)) assert max(sub.end - sub.start for sub in parser.subs_) <= timedelta(seconds=max_seconds)