删除/覆盖导入

Question

我正在尝试使用unittest 为介绍 CS 课程设置评分脚本。本质上，学生提交了一个 python 文件student.py，其中包含一些通常相互依赖的函数（这意味着func3() 可以在其计算中使用func1()）。

我正在通过将student.func1 的输出与correct.func1 的输出进行比较来为每种方法编写单元测试，这种方法已知是正确的实现（来自文件correct.py）。

例如，假设func2 在其计算中使用func1。 因此，无论是默认情况下还是student.func1 未通过某些测试，我都想用correct.func1 覆盖student.func1，因此student.func2 使用已知的正确实现（因此默认情况下不仅仅是错误的）。我怎么能这样做呢？ setUp() 和 tearDown() 似乎与我想要的相似，但我不知道如何在 python 中“取消导入”模块，并且到目前为止还没有找到任何相关资源。

我对@9​​87654336@ 包含类和func1、func2 是特定类的方法以及func1 和func2 只是在student.py 中一般定义的情况感兴趣.

Answer 1

最简单的方法是将student 导入到您的模块中，然后在测试失败时捕获AssertionError，并用您自己的好代码替换学生模块中的坏代码：

import student

import unittest

def safe_f1():
    print("Safe f1")
    return 1

class TestSomething(unittest.TestCase):

    def test_f1(self):
        try:
            self.assertEqual(student.func1(), 1)
        except AssertionError:
            student.func1 = safe_f1
            raise


    def test_f2(self):
        self.assertEqual(student.func2(), 2)

这是一个失败/有效的虚拟 student.py：

def func1():
    print("Bad f1")
    return 2

def func2():
    return func1() + 1
    return 2

当我运行它时，我得到：

$ python -m unittest test.py
Bad f1
FSafe f1
.
======================================================================
FAIL: test_f1 (test.TestSomething)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/austin/Code/so/test.py", line 13, in test_f1
    self.assertEqual(student.func1(), 1)
AssertionError: 2 != 1

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

FAILED (failures=1)