Question

我有一个函数应该将对象属性设置为PIL Image实例：

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

assertIsInstance（）针对多个类名

1 个答案: