sync with unit tests

Author: FindHao

.ci/install-project.sh

@@ -28,6 +28,6 @@ pip install -e ".[test]"
 # Verify installation
 echo "Verifying installation..."
 python -c "import tritonparse; print(f'tritonparse installed successfully')"
-python -c "import pytest; print(f'pytest version: {pytest.__version__}')"
+python -c "import coverage; print(f'coverage version: {coverage.__version__}')"
 
 echo "Project installation completed successfully!" 

.ci/run-tests.sh

@@ -29,34 +29,66 @@ conda activate "$CONDA_ENV"
 # export TORCHINDUCTOR_FX_GRAPH_CACHE=0
 # export TRITONPARSE_DEBUG=1
 
-# Build pytest command
-PYTEST_CMD="python -m unittest tests.test_tritonparse"
-
-# Add verbose flag
-if [ "$VERBOSE" = "true" ]; then
-    PYTEST_CMD="$PYTEST_CMD -v"
-fi
-
-# Add coverage if requested
-if [ "$COVERAGE" = "true" ]; then
-    PYTEST_CMD="$PYTEST_CMD --cov=tritonparse --cov-report=xml"
-fi
-
-# Run tests based on type
+# Build unittest command based on test type
 case "$TEST_TYPE" in
     "cpu")
         echo "Running CPU tests only..."
-        $PYTEST_CMD -m "not cuda"
+        if [ "$COVERAGE" = "true" ]; then
+            echo "Running with coverage..."
+            if [ "$VERBOSE" = "true" ]; then
+                coverage run -m unittest tests.test_tritonparse.TestTritonparseCPU -v
+            else
+                coverage run -m unittest tests.test_tritonparse.TestTritonparseCPU
+            fi
+            coverage report
+            coverage xml
+        else
+            if [ "$VERBOSE" = "true" ]; then
+                python -m unittest tests.test_tritonparse.TestTritonparseCPU -v
+            else
+                python -m unittest tests.test_tritonparse.TestTritonparseCPU
+            fi
+        fi
         ;;
     "cuda")
         echo "Running CUDA tests only..."
         export CUDA_VISIBLE_DEVICES=0
-        $PYTEST_CMD -m cuda
+        if [ "$COVERAGE" = "true" ]; then
+            echo "Running with coverage..."
+            if [ "$VERBOSE" = "true" ]; then
+                coverage run -m unittest tests.test_tritonparse.TestTritonparseCUDA -v
+            else
+                coverage run -m unittest tests.test_tritonparse.TestTritonparseCUDA
+            fi
+            coverage report
+            coverage xml
+        else
+            if [ "$VERBOSE" = "true" ]; then
+                python -m unittest tests.test_tritonparse.TestTritonparseCUDA -v
+            else
+                python -m unittest tests.test_tritonparse.TestTritonparseCUDA
+            fi
+        fi
         ;;
     "all")
         echo "Running all tests..."
         export CUDA_VISIBLE_DEVICES=0
-        $PYTEST_CMD
+        if [ "$COVERAGE" = "true" ]; then
+            echo "Running with coverage..."
+            if [ "$VERBOSE" = "true" ]; then
+                coverage run -m unittest tests.test_tritonparse -v
+            else
+                coverage run -m unittest tests.test_tritonparse
+            fi
+            coverage report
+            coverage xml
+        else
+            if [ "$VERBOSE" = "true" ]; then
+                python -m unittest tests.test_tritonparse -v
+            else
+                python -m unittest tests.test_tritonparse
+            fi
+        fi
         ;;
     *)
         echo "Unknown test type: $TEST_TYPE"

pyproject.toml

@@ -11,9 +11,7 @@ dependencies = [
 
 [project.optional-dependencies]
 test = [
-    "pytest>=7.0.0",
-    "pytest-cov>=4.0.0",
-    "pytest-xdist>=3.0.0",
+    "coverage>=7.0.0",
 ]
 
 [tool.setuptools.packages.find]
@@ -26,20 +24,5 @@ line-length = 88
 line-length = 88
 target-version = ["py310"]
 
-[tool.pytest.ini_options]
-testpaths = ["tests"]
-python_files = ["test_*.py"]
-python_classes = ["Test*"]
-python_functions = ["test_*"]
-addopts = [
-    "-v",
-    "--tb=short",
-    "--strict-markers",
-    "--disable-warnings",
-]
-markers = [
-    "cuda: mark test as requiring CUDA",
-]
-
 [project.urls]
 "Homepage" = "https://github.com/pytorch-labs/tritonparse"