使用 unittest 和 argparse 测试参数解析

Question

我正在为我的 python 脚本写一些unit tests。该脚本允许用户覆盖参数的值。

python paramParser.py -d C:\\Path\\to\\yourfile\\xxx.m -p 'xxxParamName' -v 20

-d 文件名，可以包含参数名称和值
-p 参数名称
-v 参数值

脚本读取文件，如果参数存在，它将获取其值并将其写入具有新值的另一个文件。

从command-line 运行它成功地覆盖了参数的值。

这是我的代码：

import unittest
import paramParser

class Tests(unittest.TestCase):

  def testOneIntValue(self):
    result = paramParser.main(["-d C:\\Path\\to\\yourfile\\xxx.m", "-p 'xxxParamName'", "-v 3"])
    self.assertTrue('xxxParamName.Value= 3', result)

  def testMultipleIntValues(self):
    result = paramParser.main(["-d C:\\Path\\to\\yourfile\\xxx.m", "-p 'xxxParamName'", "-v 3"])
    self.assertTrue('[ xxxParamName.Value = [ 1 2 3 ]', result)

  def testTrueBoolean(self):
    result = paramParser.main(["-d C:\\Path\\to\\yourfile\\xxx.m", "-p 'xxxParamName'", "-v true"])
    self.assertTrue('xxxParamName.Value = true', result)

if __name__ == '__main__':
    unittest.main()

测试总是通过，知道我做错了什么吗？

编辑 我试着和这个人一样做：argparse fails when called from unittest test

def testOneValue(self):
    parsed = paramParser.main(["-d","C:\\Path\\to\\yourfile\\xxx.m", "-p", "xxxParamName", "-v", "3"])
    self.assertEqual(parsed["d"], "C:\\Path\\to\\yourfile\\xxx.m")
    self.assertEqual(parsed["p"], "xxxParamName")
    self.assertEqual(parsed["v"], "3")

虽然他的传球（借助 [] 并分隔参数）我的传球没有。我收到此错误：

TypeError: 'NoneType' object has no attribute '__getitem__'

只是关于上一期的快速更新。我忘了在我的 def main() 中添加一个返回值。测试现在正在运行。

Answer 1

在assertTrue 语句中，您传递了一个非空string，它的计算结果始终为True。

如果paramParser.main 返回它的参数，你应该这样做：

def testOneIntValue(self):
    result = paramParser.main(["-d C:\\Path\\to\\yourfile\\xxx.m", "-p 'xxxParamName'", "-v 3 "])
    self.assertEqual(result.xxxParamName, 3)   # access xxxParamName here the way you pass it

另外，您可能想了解unittest 模块并查看这个问题 - How do you write tests for argparse?。