def test_same_encoding(encoding): parser = SrtParser(encoding=encoding) offseter = SrtOffseter(1) pipe = make_pipeline(parser, offseter) pipe.fit(BytesIO(fake_srt)) assert parser.subs_.encoding == encoding assert offseter.subs_.encoding == parser.subs_.encoding assert offseter.subs_.set_encoding('same').encoding == encoding assert offseter.subs_.set_encoding('utf-8').encoding == 'utf-8'
def test_offset(offset): parser = SrtParser() offseter = SrtOffseter(offset) pipe = make_pipeline(parser, offseter) pipe.fit(BytesIO(fake_srt)) for sub_orig, sub_offset in zip(parser.subs_, offseter.subs_): assert abs(sub_offset.start.total_seconds() - sub_orig.start.total_seconds() - offset) < 1e-6 assert abs(sub_offset.end.total_seconds() - sub_orig.end.total_seconds() - offset) < 1e-6