assertIsInstance()针对多个类名

Question

我有一个函数应该将对象属性设置为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(),但是对结果实例的内省给出了一个不同的类.名称取决于加载的图像的文件类型.我在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的所有帮助.

Answer 1

使用父类 ; 你不关心这是多少子类中的哪一个,就好像它是一个图像文件.

第二个参数assertIsInstace()必须是类本身,而不是类名:

from PIL import ImageFile

self.assertIsInstance(SimpleExample.img, ImageFile.ImageFile)

或者你可以更进一步并断言它是一个PIL.Image.Image实例,并不关心它是来自特定文件还是以其他方式生成:

from PIL import Image

self.assertIsInstance(SimpleExample.img, Image.Image)

如果你确实需要针对多个类进行测试,isinstance()(并且通过扩展assertIsInstance())也接受一个类元组作为第二个参数:

>>> foo = 'bar'
>>> isinstance(foo, (str, int, float))
True