|
""" |
|
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, |
|
) |
|
|