def fail_unless_equal_main_with_this_argv(self, runfile='', argv=[], expected=[]): # catch output output = [] run_command = give_mocked_run_command(output) # call main func without arguments main(runfile=runfile, argv=argv, run_command=run_command) self.failUnlessEqual(expected, output)
def test_with_real_run_command(self): cmd = '''"%s" -c "f = file('x', 'w'); f.write('xxx'); f.close();"''' % sys.executable argv = [cmd, 'a', 'b', 'c'] main(argv=argv, quiet=True) self.failUnlessEqual(['x'], os.listdir('./a/')) self.failUnlessEqual(['x'], os.listdir('./b/')) self.failUnlessEqual(['x'], os.listdir('./c/'))
#!/usr/bin/env python import sys, os from os import path base = path.abspath(path.join(path.dirname(__file__), path.pardir)) kvipy = path.join(base, 'kvipytools') if kvipy in sys.path: sys.path.remove(kvipy) sys.path.insert(0, kvipy) os.chdir(base) from kvipytools.run import main main()