blob: 0679307397ba87673de37a39064612e6424aaa43 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Test runner for sqlparse."""
import hotshot
import optparse
import os
import sys
import unittest
sys.path.insert(1, os.path.join(os.path.dirname(__file__), '../'))
parser = optparse.OptionParser()
parser.add_option('-P', '--profile',
help='Create hotshot profile.',
action='store_true', default=False)
def main(args):
"""Create a TestSuite and run it."""
loader = unittest.TestLoader()
suite = unittest.TestSuite()
fnames = [os.path.split(f)[-1] for f in args]
for fname in os.listdir(os.path.dirname(__file__)):
if (not fname.startswith('test_') or not fname.endswith('.py')
or (fnames and fname not in fnames)):
continue
modname = os.path.splitext(fname)[0]
mod = __import__(os.path.splitext(fname)[0])
suite.addTests(loader.loadTestsFromModule(mod))
unittest.TextTestRunner(verbosity=2).run(suite)
if __name__ == '__main__':
opts, args = parser.parse_args()
if opts.profile:
prof = hotshot.Profile("sqlparse.prof")
prof.runcall(main, args)
prof.close()
else:
main(args)
|
