modelscope/tests/preprocessors/test_image.py

# Copyright (c) Alibaba, Inc. and its affiliates.

import unittest

from PIL import Image

from modelscope.preprocessors import load_image


class ImagePreprocessorTest(unittest.TestCase):

    def test_load(self):
        img = load_image('data/test/images/image_matting.png')
        self.assertTrue(isinstance(img, Image.Image))
        self.assertEqual(img.size, (948, 533))


if __name__ == '__main__':
    unittest.main()