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)))