为单元测试传递命令行参数

时间:2018-06-28 14:34:38

标签: python unit-testing

我的__main__.py文件中包含以下代码:

from argparse import ArgumentParser
from pwpcore.job import Job, job
from GlobalMacro.GlobalMacroGen import GlobalMacroGen

parser = ArgumentParser(
    description=GlobalMacroGen.__doc__,
    conflict_handler="resolve",
    parents=[job.parser])
parser.add_argument('--wikiPage', default="http://wiki.noco.corp/display/PD/Global+Macro+Generator")
parser.add_argument('--notionalSymbols', nargs='*')
parser.add_argument('--positionDate', required=True)
parser.add_argument('--fundCode', required=True)

args = parser.parse_args()

with Job(GlobalMacroGen.__name__, args) as j:
    loader = GlobalMacroGen(j, args.notionalSymbols, args.positionDate, args.fundCode, args.outputDir)
    loader.load_file()

我的GlobalMacroGen模块使用这些参数初始化,然后在某些函数中使用它们。如何在我的Unittest文件中传递虚拟参数?

我有以下内容:

导入单元测试 从pwpcore.job.job导入Job,解析器 从GlobalMacro.GlobalMacroGen导入GlobalMacroGen

class Test_global_macro_gen(unittest.TestCase):
    def setUp(self):
        self.job = Job('Unit Test', args=self.args)
        self.generator = GlobalMacroGen(self.job, )

    def test_

0 个答案:

没有答案