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