unittest 引发异常 python

Question

我有以下内容来查看数据帧是否为空：

def emptyframecheck(data_frame):
if data_frame.empty:
raise Exception('An empty dataframe was passed.')

我想对此进行单元测试，但是当我运行它时：

def test_emptyframecheck():
# Given
workitem_df = pd.read_csv(os.path.join(HERE, 'test_csv', 'test.csv'))
# When
emptyframecheck(workitem_df)
# Then
assertTrue('An empty dataframe was passed' in emptyframecheck.exception)

我得到一个未定义的全局名称assertTrue。workitem_df只是我正在传递的一个空框架。我缺少什么或更好的方法吗？

Answer 1

此错误的意思是：未定义assertTrue。你进口过assertTrue吗？如果您导入了测试模块但没有导入框架，则必须执行以下操作：

testing_module.assertTrue()

如果要直接使用"assertTrue"，则应在测试文件的顶部编写以下内容：

from testing_module import assertTrue