63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
|
import unittest
|
||
|
|
import src.fsub.fsub as fsub
|
||
|
|
import shutil
|
||
|
|
import os
|
||
|
|
import inspect
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
class TestFsub(unittest.TestCase):
|
||
|
|
samples = Path('tests/samples')
|
||
|
|
maxDiff = None
|
||
|
|
|
||
|
|
def run_on(self, args, sample, ofile):
|
||
|
|
ifile = inspect.stack()[1][3] + '.srt'
|
||
|
|
|
||
|
|
sample = str(self.samples / sample) + '.srt'
|
||
|
|
shutil.copy(sample, ifile)
|
||
|
|
args.append(ifile)
|
||
|
|
|
||
|
|
fsub.run(args)
|
||
|
|
|
||
|
|
out = open(ifile)
|
||
|
|
result = out.read()
|
||
|
|
out.close()
|
||
|
|
|
||
|
|
ofile = str(self.samples / ofile) + '.srt'
|
||
|
|
cmp_file = open(ofile)
|
||
|
|
cmp = cmp_file.read()
|
||
|
|
cmp_file.close()
|
||
|
|
|
||
|
|
self.assertEqual(result, cmp)
|
||
|
|
os.remove(ifile)
|
||
|
|
|
||
|
|
def test_cleaned(self):
|
||
|
|
args = ['-f', str(self.samples / 'blacklist')]
|
||
|
|
self.run_on(args, 'sample1', 'sample1-cleaned')
|
||
|
|
|
||
|
|
def test_stripped(self):
|
||
|
|
self.run_on(['-n'], 'sample1', 'sample1-stripped')
|
||
|
|
|
||
|
|
def test_cleaned_stripped(self):
|
||
|
|
args = ['-c', '-f', str(self.samples / 'blacklist'), '-n']
|
||
|
|
self.run_on(args, 'sample1', 'sample1-cleaned-stripped')
|
||
|
|
|
||
|
|
def test_cleaned_stripped_shifted_1h(self):
|
||
|
|
args = ['-c',
|
||
|
|
'-f', str(self.samples / 'blacklist'),
|
||
|
|
'-n',
|
||
|
|
'-s', '3600000']
|
||
|
|
self.run_on(args, 'sample1', 'sample1-cleaned-stripped-shifted-1h')
|
||
|
|
|
||
|
|
def test_shifted_minus_1h(self):
|
||
|
|
args = ['-s', '-3600000']
|
||
|
|
self.run_on(args, 'sample1', 'sample1-shifted-minus-1h')
|
||
|
|
|
||
|
|
def test_shifted_minus_52s(self):
|
||
|
|
args = ['-s', '-52000']
|
||
|
|
self.run_on(args, 'sample1', 'sample1-shifted-minus-52s')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|