InvokeAI/tests/test_path.py

"""
Not really a test, but a way to verify that the paths are existing
and fail early if they are not.
"""

import pathlib
import unittest
from os import path as osp

from PIL import Image

import invokeai.app.assets.images as image_assets
import invokeai.configs as configs


class ConfigsTestCase(unittest.TestCase):
    """Test the configuration related imports and objects"""

    def get_configs_path(self) -> pathlib.Path:
        """Get the path of the configs folder"""
        configs_path = pathlib.Path(configs.__path__[0])
        return configs_path

    def test_configs_path(self):
        """Test that the configs path is correct"""
        TEST_PATH = str(self.get_configs_path())
        assert TEST_PATH.endswith(str(osp.join("invokeai", "configs")))

    def test_caution_img(self):
        """Verify the caution image"""
        caution_img = Image.open(osp.join(image_assets.__path__[0], "caution.png"))
        assert caution_img.width == int(500)
        assert caution_img.height == int(441)
        assert caution_img.format == str("PNG")


if __name__ == "__main__":
    unittest.main(
        verbosity=2,
    )