使用argparse进行Python单元测试

Question

我正在尝试将unittest与使用argparse模块的程序一起使用并遇到一些困难。我引用this helpful post作为起点，但我显然仍然遗漏了一些东西。

这是基本程序：

#argTest.py
import sys
import argparse


class Thingy:
    def __init__(self, name):
        self.name = name


    def parse_args(args):
        parser = argparse.ArgumentParser(description='description here')
        parser.add_argument('-v', '--version', action='version', version='%(prog)s 0.1')
        parser.add_argument('-a', '--arg1', required=True, help='this is for arg1')
        parser.add_argument('-b', '--arg2', required=True, help='this is for arg2')
        return parser.parse_args()


def main():
    parser = Thingy.parse_args(sys.argv[1:])
    print('the args are: {}'.format(parser))
    if parser.arg1:
        print('the value of arg1 is : {}'.format(parser.arg1))
    if parser.arg2:
        print('the value of arg2 is : {}'.format(parser.arg2))


if __name__ == '__main__':
    main()

运行此：

python argTest.py --arg1 asdf --arg2 qwer

预期产出的结果：

the args are: Namespace(arg1='asdf', arg2='qwer')
the value of arg1 is : asdf
the value of arg2 is : qwer

现在这里是简单的单元测试程序：

#test/test_argTest.py
import unittest
from argTest import Thingy


def test_parser(self):
    parser = Thingy.parse_args(['--arg1'])
    self.assertTrue(parser.arg1,'asdf')


if __name__ == '__main__':
    main()

运行此：

python -m unittest -v test/test_argTest.py --arg1 asdf --arg2 qwer

结果如下：

python -m unittest -v test/test_argTest.py --arg1 asdf --arg2 qwer
usage: python -m unittest [-h] [-v] [-q] [--locals] [-f] [-c] [-b]
                          [tests [tests ...]]
python -m unittest: error: unrecognized arguments: --arg1 asdf --arg2 qwer

有人可以指出我如何进行这些测试吗？

感谢。

更新＃1

以下是基于以下有用建议的更新的unittest程序，但仍然缺少某些内容。

#import unittest
from argTest import Thingy

class TestThingys(unittest.TestCase):
    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_parser(self):
        argv1 = ['--arg1', 'asdf', '--arg2', 'qwer']
        parser = Thingy().parse_args(argv1)
        self.assertTrue(parser.arg1,'asdf')

if __name__ == '__main__':
    main()

运行此：

python -m unittest -v test/test_argTest.py

结果如下：

test_isupper (test.test_argTest.TestThingys) ... ok
test_parser (test.test_argTest.TestThingys) ... ERROR

======================================================================
ERROR: test_parser (test.test_argTest.TestThingys)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/username/scripts/python/argTest/test/test_argTest.py", line 12, in test_parser
    parser = Thingy().parse_args(argv1)
TypeError: __init__() missing 1 required positional argument: 'name'

----------------------------------------------------------------------
Ran 2 tests in 0.001s

FAILED (errors=1)

更新＃2

#test/test_argTest.py
import unittest
from argTest import Thingy

class TestThingys(unittest.TestCase):
    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_parser(self):
        argv1 = ['--arg1', 'asdf', '--arg2', 'qwer']
        parser = Thingy('name').parse_args(argv1)
        self.assertEquals(parser.arg1,'asdf')

if __name__ == '__main__':
    main()

运行此：

python -m unittest -v test/test_argTest.py

结果如下：

test_isupper (test.test_argTest.TestThingys) ... ok
test_parser (test.test_argTest.TestThingys) ... ERROR

======================================================================
ERROR: test_parser (test.test_argTest.TestThingys)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/username/scripts/python/argTest/test/test_argTest.py", line 12, in test_parser
    parser = Thingy('name').parse_args(argv1)
TypeError: parse_args() takes 1 positional argument but 2 were given

----------------------------------------------------------------------
Ran 2 tests in 0.001s

FAILED (errors=1)

Answer 1

在argv

中使用class TestThingy

def test_parser(self):
    argv1 = ['--arg1', 'asdf', '--arg2', 'qwer']
    parser = Thingy('name').parse_args(argv1)
    self.assertEquals(parser.arg1,'asdf')

    argv2 = ['--triiger_exception`, 'asdf`]
    with self.assertRaise(Exception):
        parser = Thingy('name').parse_args(argv2)