tests: setting up more tests

Author: UnicoLab

Makefile

@@ -13,8 +13,15 @@ help:
 	@echo "  make lint             Run linting checks"
 	@echo "  make type-check       Run type checking with mypy"
 	@echo "  make test             Run tests"
+	@echo "  make test-unit        Run unit tests only"
+	@echo "  make test-integration Run integration tests only"
+	@echo "  make test-performance Run performance tests only"
+	@echo "  make test-async       Run async tests only"
 	@echo "  make test-parallel    Run tests in parallel"
 	@echo "  make test-coverage    Run tests with coverage report"
+	@echo "  make test-fast        Run fast tests only (exclude slow)"
+	@echo "  make test-all         Run all tests including performance"
+	@echo "  make test-ci          Run CI test suite"
 	@echo ""
 	@echo "Documentation:"
 	@echo "  make docs             Build documentation"
@@ -62,6 +69,22 @@ test:
 	@echo "Running tests..."
 	pytest tests/ -v
 
+test-unit:
+	@echo "Running unit tests..."
+	pytest tests/ -v -m "unit"
+
+test-integration:
+	@echo "Running integration tests..."
+	pytest tests/ -v -m "integration"
+
+test-performance:
+	@echo "Running performance tests..."
+	pytest tests/ -v -m "performance"
+
+test-async:
+	@echo "Running async tests..."
+	pytest tests/ -v -m "async"
+
 test-parallel:
 	@echo "Running tests in parallel..."
 	pytest tests/ -v -n auto --dist loadgroup
@@ -71,6 +94,18 @@ test-coverage:
 	pytest tests/ -v --cov=graphflow --cov-report=html --cov-report=term-missing
 	@echo "Coverage report generated in htmlcov/index.html"
 
+test-fast:
+	@echo "Running fast tests only..."
+	pytest tests/ -v -m "not slow"
+
+test-all:
+	@echo "Running all tests including performance..."
+	pytest tests/ -v
+
+test-ci:
+	@echo "Running CI test suite..."
+	pytest tests/ -c pytest-fast.ini --cov=graphflow --cov-report=xml --cov-report=html --timeout=180
+
 # Documentation
 docs:
 	@echo "Building documentation..."

UNICOLAB_ADAPTATION_COMPLETE.md

@@ -69,6 +69,7 @@ dev = [
     "pytest-cov>=4.1.0",
     "pytest-xdist>=3.3.0",
     "pytest-asyncio>=0.21.0",
+    "pytest-timeout>=2.1.0",
     "black>=23.7.0",
     "isort>=5.12.0",
     "mypy>=1.5.0",

pyproject.toml

@@ -0,0 +1,23 @@
+[pytest]
+testpaths = tests
+python_files = test_*.py
+python_classes = Test*
+python_functions = test_*
+addopts =
+    -v
+    --tb=short
+    --strict-markers
+    --disable-warnings
+    --cov=graphflow
+    --cov-report=term-missing
+    --cov-report=html
+    --cov-report=xml
+    --asyncio-mode=auto
+markers =
+    slow: marks tests as slow (deselect with '-m "not slow"')
+    integration: marks tests as integration tests
+    unit: marks tests as unit tests
+    performance: marks tests as performance tests (benchmarks)
+    async: marks tests as async tests
+    parallel: marks tests that can run in parallel
+asyncio_mode = auto

pytest-simple.ini

@@ -14,9 +14,13 @@ addopts =
     --cov-report=xml
     -n auto
     --dist loadgroup
+    --asyncio-mode=auto
 markers =
     slow: marks tests as slow (deselect with '-m "not slow"')
     integration: marks tests as integration tests
     unit: marks tests as unit tests
+    performance: marks tests as performance tests (benchmarks)
+    async: marks tests as async tests
     parallel: marks tests that can run in parallel
+asyncio_mode = auto