Python：单元测试中的assertRaises错误...未捕获异常

Question

我编写了一个单元测试来强制抛出异常。抛出异常，但我的单元测试语句由于某种原因没有捕获它，并且意外失败。

这是单元测试：

def test900_001_ShouldRaiseExceptionDuplicateID(self):
    hist = projecthistory.ProjectHistory()
    myProject = project.Project(id = 42, locR = 10, locP = 15, locA = 30, eP = 200, eA= 210)
    hist.addProject(myProject)
    myProject2 = project.Project(id = 42, locR = 15, locP = 25, locA = 40, eP = 300, eA = 410)
    self.assertRaises(ValueError, projecthistory.ProjectHistory, hist.addProject(myProject2))

这是与此相关的代码：

def addProject(self, proj):

    duplicate = False
    checkId = proj.getId()

    #check to see if that id is already in the container if so, raise ValueError
    #append project to container
    for project in self.theContainer:
        if (project.getId() == checkId):
            duplicate = True
            break
    if(duplicate == False):
        self.theContainer.append(proj)
    else:
        raise ValueError("ProjectHistory.addProject: Duplicate ID found. Project not added to repository.")

    return len(self.theContainer)

基本上，项目被添加到名为 theContainer 的列表中。但是，如果两个 ID 相同，则不会添加副本。通过在要添加的单元测试中强制添加两个具有相同 ID 的项目，会引发异常。

这是我得到的回溯：

Traceback (most recent call last):

  File "C:\Users\blah\workspace\blahID\CA06\test\projecthistoryTest.py", line 46, in test900_001_ShouldRaiseExceptionDuplicateID
    self.assertRaises(ValueError, projecthistory.ProjectHistory, hist.addProject(myProject2))

  File "C:\Users\blah\workspace\blahID\CA06\prod\projecthistory.py", line 38, in addProject
    raise ValueError("ProjectHistory.addProject: Duplicate ID found. Project not added to repository.")
ValueError: ProjectHistory.addProject: Duplicate ID found. Project not added to repository.

问题可能出在 assertRaises 中的第三个参数上吗？ (hist.addProject(myProject2))

Answer 1

您的怀疑是正确的，问题在于调用hist.addProject()。

你写道：

self.assertRaises(ValueError, projecthistory.ProjectHistory,
    hist.addProject(myProject2))

提出了一个ValueError。但它在

hist.addProject(myProject2)

回溯告诉你。所以assertRaises 从来没有真正被调用过，因为在它被调用之前就引发了异常。

考虑它的方式是assertRaises 只有在它确实设法被调用时才能捕获异常。如果准备参数的行为引发了异常，那么assertRaises 甚至不会运行，因此无法捕获任何内容。

如果您希望在调用 addProject() 时出现异常，只需更改您的断言：

self.assertRaises(ValueError, hist.addProject, myProject2)

或者您可以使用 lambda 推迟对 hist.addProject() 的调用：

self.assertRaises(ValueError, 
    lambda: projecthistory.ProjectHistory(hist.addProject(myProject2)))