diff --git a/tests/test_linux_client.py b/tests/test_linux_client.py index 44d9c46..f06ce07 100644 --- a/tests/test_linux_client.py +++ b/tests/test_linux_client.py @@ -8,6 +8,7 @@ import warnings warnings.filterwarnings("ignore") sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src"))) import Linux_Client import pgpy diff --git a/tests/test_server.py b/tests/test_server.py index d1c0bb2..ac884b7 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -11,8 +11,9 @@ from datetime import datetime, timezone, timedelta warnings.filterwarnings("ignore") -# Ensure parent directory is in path to import Server +# Ensure parent directory and src directory are in path to import Server sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src"))) import Server import pgpy diff --git a/tests/test_win_client.py b/tests/test_win_client.py index 5431373..ab06652 100644 --- a/tests/test_win_client.py +++ b/tests/test_win_client.py @@ -8,6 +8,7 @@ import warnings warnings.filterwarnings("ignore") sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src"))) import Win_Client import pgpy