diff --git a/tests/test_path.py b/tests/test_path.py index 1c3276b659..e99420d8ee 100644 --- a/tests/test_path.py +++ b/tests/test_path.py @@ -1,32 +1,48 @@ -import pathlib +""" +Not really a test, but a way to verify that the paths are existing +and fail early if they are not. +""" import unittest +import pathlib from os import path as osp +from PIL import Image -import invokeai.configs as configs import invokeai.frontend.dist as frontend +import invokeai.configs as configs +import invokeai.assets.web as assets_web -def get_frontend_path() -> pathlib.Path: - """Get the path of the frontend dist folder""" - return pathlib.Path(frontend.__path__[0]) +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 get_configs_path() -> pathlib.Path: - """Get the path of the configs folder""" - return pathlib.Path(configs.__path__[0]) + def get_frontend_path(self) -> pathlib.Path: + """Get the path of the frontend dist folder""" + return pathlib.Path(frontend.__path__[0]) + 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_frontend_path(): - """Test that the frontend path is correct""" - TEST_PATH = str(get_frontend_path()) - assert TEST_PATH.endswith(osp.join("invokeai", "frontend", "dist")) + def test_frontend_path(self): + """Test that the frontend path is correct""" + FRONTEND_PATH = str(self.get_frontend_path()) + assert FRONTEND_PATH.endswith(osp.join("invokeai", "frontend", "dist")) - -def test_configs_path(): - """Test that the configs path is correct""" - TEST_PATH = str(get_configs_path()) - assert TEST_PATH.endswith(osp.join("invokeai", "configs")) + def test_caution_img(self): + """Verify the caution image""" + caution_img = Image.open(osp.join(assets_web.__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() + unittest.main( + verbosity=2, + )