我有一个函数应该将对象属性设置为PIL图像实例:
from PIL import Image
class SimpleExample:
def __init__(self):
self.img = self.load_image()
def load_image():
self.img = Image.open(image)
为了测试load_image()的正确执行,我想使用unittest.TestCase.assertIsInstance(),但是对结果实例的内省给出了一个不同的class.name,具体取决于加载的图像的文件类型.我在unittest中找不到assertIsInstanceFromModule()的任何内容,所以这似乎是最能反映我真正想要测试的断言助手.
我想出了几个解决方案,但对其中任何一个都不满意:
使用循环
生成所有可能的类名列表,迭代列表,测试每个成员是否相等,然后assertIsInstance为适当的列表成员.这是尴尬的,单声道的,我当然不会为此感到骄傲.它确实有效,它使用我认为最有意义的断言.
simplified_names = ['JpegImageFile', 'PngImageFile']
for i, name in enumerate(simplified_names):
if SimpleExample.img.__class__.__name__ == name:
TestCase.assertIsInstance(SimpleExample.img, simplified_names[i])
使用不同的断言助手
生成所有可能类名的列表,并声明生成的对象的类名在该列表中.它很有用,很简单,但我不喜欢我实际上在这里断言的内容.
simplified_names = ['JpegImageFile', 'PngImageFile']
TestCase.assertIn(SimpleExample.img.__class__.__name__, simplified_names)
有没有人对如何断言多个可能的类名有任何其他想法?
我是否应该从不同的方向接近测试并使用不同的断言?
这是我第一次涉足unittest.我应该回去只用print来测试我的代码吗?
~~~~~
明显的答案是……
测试父类.或者,我可以注意到每次看到以下错误时都会向我提出更好的选项:
TypeError: isinstance() arg 2 must be a type or tuple of types
强调“或元组”.
感谢Martijn Pieters的所有帮助.