Fix "pytest cannot find module": rm __initL__.py, manual sys.path, add

pythonpath
antoinetran · Jan 3, 2024 · 91d8d80 · 91d8d80
1 parent 80ff102
commit 91d8d80
Show file tree

Hide file tree

Showing 8 changed files with 1 addition and 25 deletions.
diff --git a/__init__.py b/__init__.py
diff --git a/pytest.ini b/pytest.ini
@@ -1,3 +1,4 @@
 # Handling DeprecationWarning 'asyncio_mode' default value
 [pytest]
 asyncio_mode = strict
+pythonpath = .
diff --git a/tests/advanced_rpc_test.py b/tests/advanced_rpc_test.py
@@ -1,10 +1,6 @@
 import os
 import sys
 
-# Add parent path to use local src as package for tests
-sys.path.append(os.path.abspath(os.path.join(
-    os.path.dirname(__file__), os.path.pardir)))
-
 import time 
 import asyncio
 from multiprocessing import Process

diff --git a/tests/basic_rpc_test.py b/tests/basic_rpc_test.py
@@ -2,10 +2,6 @@
 import os
 import sys
 
-# Add parent path to use local src as package for tests
-sys.path.append(os.path.abspath(os.path.join(
-    os.path.dirname(__file__), os.path.pardir)))
-
 import asyncio
 from multiprocessing import Process
 

diff --git a/tests/binary_rpc_test.py b/tests/binary_rpc_test.py
@@ -2,11 +2,6 @@
 import os
 import sys
 
-# Add parent path to use local src as package for tests
-sys.path.append(
-    os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir))
-)
-
 import json
 from multiprocessing import Process
 

diff --git a/tests/custom_methods_test.py b/tests/custom_methods_test.py
@@ -12,10 +12,6 @@
 import os
 import sys
 
-# Add parent path to use local src as package for tests
-sys.path.append(os.path.abspath(os.path.join(
-    os.path.dirname(__file__), os.path.pardir)))
-
 
 # Configurable
 PORT = int(os.environ.get("PORT") or "9000")

diff --git a/tests/fast_api_depends_test.py b/tests/fast_api_depends_test.py
@@ -3,10 +3,6 @@
 
 from websockets.exceptions import InvalidStatusCode
 
-# Add parent path to use local src as package for tests
-sys.path.append(os.path.abspath(os.path.join(
-    os.path.dirname(__file__), os.path.pardir)))
-
 from multiprocessing import Process
 
 import pytest

diff --git a/tests/trigger_flow_test.py b/tests/trigger_flow_test.py
@@ -5,10 +5,6 @@
 import os
 import sys
 
-# Add parent path to use local src as package for tests
-sys.path.append(os.path.abspath(os.path.join(
-    os.path.dirname(__file__), os.path.pardir)))
-
 from fastapi_websocket_rpc.utils import gen_uid
 from fastapi_websocket_rpc.websocket_rpc_endpoint import WebsocketRPCEndpoint
 from fastapi_websocket_rpc.websocket_rpc_client import WebSocketRpcClient