2 from colorama import Fore
3 from helpers import deflate, inflate, run_cmd
5 outDirDefault = 'test-outs'
6 testDirDefault = 'test-files'
11 @param tFile- required; the full path to the file to run
12 @param level- optional (default: all); the compression level [1-9]
13 @param delete- optional (default: True); whether to delete the gzipped files
14 @return True if all tests passed; False if at least one test failed
16 def runTest(tFile, level=None, delete=True, outDir=outDirDefault):
19 for x in range(1, 10):
20 if runTest(tFile, x, delete) == False:
25 # make the test-outs directory
31 out1 = os.path.join(outDir, '%(file)s.%(level)d.deflate' % {'file': os.path.basename(tFile), 'level' : level})
32 out2 = os.path.join(outDir, '%(file)s.%(level)d.out.deflate' % {'file': os.path.basename(tFile), 'level' : level})
34 outData = deflate(tFile, outfile=out1, level=level)
35 run_cmd('../bin/deflate.js --level %(level)d --file %(file)s --output %(output)s' % {'level' : level, 'file' : tFile, 'output' : out2})
37 result = run_cmd('diff %(file1)s %(file2)s' % {'file1' : out1, 'file2' : out2})
38 if result['returncode'] == 0:
39 status = Fore.GREEN + 'PASSED' + Fore.RESET
42 status = Fore.RED + 'FAILED' + Fore.RESET
44 print 'Level %(level)d: %(status)s' % {'level' : level, 'status' : status}
53 Runs all tests on the given level. This iterates throuth the testDir directory defined above.
55 @param level- The level to run on [1-9] (default: None, runs on all levels all)
56 @param delete- Whether to delete output files after the test is run
57 @return True if all levels passed, False if at least one failed
59 def runAll(level=None, delete=True, testDir=testDirDefault, outDir=outDirDefault):
60 # make the test-outs directory
67 for tFile in os.listdir(testDir):
68 fullPath = os.path.join(testDir, tFile)
70 print Fore.YELLOW + tFile + Fore.RESET
72 if runTest(fullPath, level, delete) == False:
77 # if we deletede all the files that were created, delete the directory