From ec1e83e912d00e25089100dcea55a290792f272c Mon Sep 17 00:00:00 2001 From: mauwii Date: Tue, 31 Jan 2023 09:06:06 +0100 Subject: [PATCH] add pytest to test path of frontend and configs --- tests/test_path.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 tests/test_path.py diff --git a/tests/test_path.py b/tests/test_path.py new file mode 100644 index 0000000000..2934d29568 --- /dev/null +++ b/tests/test_path.py @@ -0,0 +1,34 @@ +import unittest +from os import path as osp +import pathlib + +import invokeai.frontend.dist as frontend +import invokeai.configs as configs + + +def get_frontend_path() -> pathlib.Path: + """Get the path of the frontend""" + return pathlib.Path(frontend.__path__[0]) + + +def get_configs_path() -> pathlib.Path: + """Get the path of the configs folder""" + return pathlib.Path(configs.__path__[0]) + + +def test_frontend_path(): + """Test that the frontend path is correct""" + # test path of the frontend + TEST_PATH = str(get_frontend_path()) + assert TEST_PATH.endswith(osp.join("invokeai", "frontend", "dist")) + + +def test_configs_path(): + """Test that the frontend path is correct""" + # test path of the frontend + TEST_PATH = str(get_configs_path()) + assert TEST_PATH.endswith(osp.join("invokeai", "configs")) + + +if __name__ == "__main__": + unittest.main()