Update

.github/workflows/test_package.yml

@@ -104,5 +104,5 @@ jobs:
       run: |
         pip install dist/*.whl
         pip install pytest pytest-asyncio
-        pytest smoke_test.py
+        pytest tests/test_basic.py
         pip uninstall -y ipybox

tests/conftest.py

@@ -0,0 +1,9 @@
+import tempfile
+
+import pytest
+
+
+@pytest.fixture(scope="module")
+async def workspace():
+    with tempfile.TemporaryDirectory() as temp_dir:
+        yield temp_dir

smoke_test.py → tests/test_basic.py

@@ -1,28 +1,19 @@
-import tempfile
-
 import pytest
 
 from ipybox import ExecutionClient, ExecutionContainer
 
 
-@pytest.fixture(scope="module")
-async def workspace():
-    with tempfile.TemporaryDirectory() as temp_dir:
-        yield temp_dir
-
-
 @pytest.fixture(scope="module")
 async def executor(workspace: str):
     async with ExecutionContainer(
         tag="ghcr.io/gradion-ai/ipybox:minimal",
         binds={workspace: "workspace"},
-        env={"TEST_VAR": "test_val"},
     ) as container:
         async with ExecutionClient(host="localhost", port=container.port) as client:
             yield client
 
 
 @pytest.mark.asyncio(loop_scope="module")
-async def test_single_command_output(executor):
+async def test_basic_functionality(executor):
     result = await executor.execute("print('Hello, world!')")
     assert result.text == "Hello, world!"

tests/ipybox_test.py → tests/test_ipybox.py

@@ -1,7 +1,6 @@
 import asyncio
 import re
 import subprocess
-import tempfile
 from pathlib import Path
 from typing import Generator
 
@@ -12,12 +11,6 @@
 from ipybox import DEFAULT_TAG, ExecutionClient, ExecutionContainer, ExecutionError
 
 
-@pytest.fixture(scope="module")
-async def workspace():
-    with tempfile.TemporaryDirectory() as temp_dir:
-        yield temp_dir
-
-
 @pytest.fixture(
     scope="module",
     params=["test-root", "test"],