python assert isinstance_python – 针对多个类名的assertIsInstance()

我有一个函数应该将对象属性设置为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的所有帮助.