refactor(tests): reorganize fitness check tests into dedicated subdirectory

Consolidate fragmented fitness check tests (test_fitness.py, test_system_fitness.py,
test_gpu_fitness.py, test_gpu_fitness_integration.py) into organized structure:

- test_fitness/conftest.py: Shared fixtures for all fitness tests
- test_fitness/test_registration.py: Fitness check framework tests
- test_fitness/test_system_checks.py: System resource check tests
- test_fitness/test_gpu_checks.py: GPU-specific check tests
- test_fitness/test_gpu_integration.py: GPU integration tests

This improves code organization and eliminates duplicate fixture definitions.

Author: deanq

tests/test_serverless/test_modules/test_fitness/__init__.py

@@ -0,0 +1 @@
+"""Fitness check system tests."""

tests/test_serverless/test_modules/test_fitness/conftest.py

@@ -0,0 +1,23 @@
+"""Shared fixtures for fitness check tests."""
+
+import pytest
+
+from runpod.serverless.modules.rp_fitness import (
+    clear_fitness_checks,
+    _reset_registration_state,
+)
+
+
+@pytest.fixture(autouse=True)
+def cleanup_fitness_checks(monkeypatch):
+    """Automatically clean up fitness checks before and after each test.
+
+    Disables auto-registration of system checks to avoid interference
+    with fitness check framework tests.
+    """
+    monkeypatch.setenv("RUNPOD_SKIP_AUTO_SYSTEM_CHECKS", "true")
+    _reset_registration_state()
+    clear_fitness_checks()
+    yield
+    _reset_registration_state()
+    clear_fitness_checks()

…verless/test_modules/test_gpu_fitness.py …_modules/test_fitness/test_gpu_checks.pytests/test_serverless/test_modules/test_gpu_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_gpu_checks.py tests/test_serverless/test_modules/test_gpu_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_gpu_checks.py

@@ -22,14 +22,6 @@
 from runpod.serverless.modules.rp_fitness import clear_fitness_checks, _fitness_checks
 
 
-@pytest.fixture(autouse=True)
-def cleanup_fitness_checks():
-    """Automatically clean up fitness checks before and after each test."""
-    clear_fitness_checks()
-    yield
-    clear_fitness_checks()
-
-
 # ============================================================================
 # Output Parsing Tests
 # ============================================================================

…_modules/test_gpu_fitness_integration.py …les/test_fitness/test_gpu_integration.pytests/test_serverless/test_modules/test_gpu_fitness_integration.py renamed to tests/test_serverless/test_modules/test_fitness/test_gpu_integration.py tests/test_serverless/test_modules/test_gpu_fitness_integration.py renamed to tests/test_serverless/test_modules/test_fitness/test_gpu_integration.py

@@ -20,18 +20,6 @@
 from runpod.serverless.modules.rp_gpu_fitness import _check_gpu_health
 
 
-@pytest.fixture(autouse=True)
-def cleanup_checks(monkeypatch):
-    """Clean fitness checks before and after each test."""
-    # Disable auto-registration of system checks for GPU fitness integration tests
-    monkeypatch.setenv("RUNPOD_SKIP_AUTO_SYSTEM_CHECKS", "true")
-    _reset_registration_state()
-    clear_fitness_checks()
-    yield
-    _reset_registration_state()
-    clear_fitness_checks()
-
-
 @pytest.fixture
 def mock_gpu_test_binary():
     """Create a temporary mock gpu_test binary that outputs success."""

…_serverless/test_modules/test_fitness.py …odules/test_fitness/test_registration.pytests/test_serverless/test_modules/test_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_registration.py tests/test_serverless/test_modules/test_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_registration.py

@@ -1,5 +1,4 @@
-"""
-Tests for the fitness check system (rp_fitness module).
+"""Tests for the fitness check system (rp_fitness module).
 
 Fitness checks are used to validate worker health at startup before handler
 initialization. Only tests registration, execution, and error handling of
@@ -14,22 +13,9 @@
     run_fitness_checks,
     clear_fitness_checks,
     _fitness_checks,
-    _reset_registration_state,
 )
 
 
-@pytest.fixture(autouse=True)
-def cleanup_fitness_checks(monkeypatch):
-    """Automatically clean up fitness checks before and after each test."""
-    # Disable auto-registration of system checks for isolated fitness check tests
-    monkeypatch.setenv("RUNPOD_SKIP_AUTO_SYSTEM_CHECKS", "true")
-    _reset_registration_state()
-    clear_fitness_checks()
-    yield
-    _reset_registration_state()
-    clear_fitness_checks()
-
-
 # ============================================================================
 # Registration Tests
 # ============================================================================
@@ -116,7 +102,6 @@ def check():
 class TestFitnessExecutionSuccess:
     """Tests for successful fitness check execution."""
 
-
     @pytest.mark.asyncio
     async def test_empty_registry_no_op(self):
         """Test that empty registry results in no-op."""
@@ -205,7 +190,6 @@ async def async_check():
 class TestFitnessExecutionFailure:
     """Tests for fitness check execution failures."""
 
-
     @pytest.mark.asyncio
     async def test_sync_check_fails(self):
         """Test that synchronous check failure causes exit."""
@@ -306,7 +290,6 @@ def check():
 class TestFitnessLogging:
     """Tests for fitness check logging behavior."""
 
-
     @pytest.mark.asyncio
     @patch("runpod.serverless.modules.rp_fitness.log")
     async def test_logs_debug_when_no_checks(self, mock_log):
@@ -395,7 +378,6 @@ def check():
 class TestFitnessClearRegistry:
     """Tests for fitness check registry cleanup."""
 
-
     def test_clear_fitness_checks(self):
         """Test that clear_fitness_checks empties the registry."""
         @register_fitness_check
@@ -428,7 +410,6 @@ def check():
 class TestFitnessIntegration:
     """Integration tests for fitness check system."""
 
-
     @pytest.mark.asyncio
     async def test_check_with_real_exception_message(self):
         """Test that real exception messages are preserved."""

…less/test_modules/test_system_fitness.py …dules/test_fitness/test_system_checks.pytests/test_serverless/test_modules/test_system_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_system_checks.py tests/test_serverless/test_modules/test_system_fitness.py renamed to tests/test_serverless/test_modules/test_fitness/test_system_checks.py

@@ -25,14 +25,6 @@
 from runpod.serverless.modules.rp_fitness import clear_fitness_checks, _fitness_checks
 
 
-@pytest.fixture(autouse=True)
-def cleanup_fitness_checks():
-    """Automatically clean up fitness checks before and after each test."""
-    clear_fitness_checks()
-    yield
-    clear_fitness_checks()
-
-
 # ============================================================================
 # Memory Check Tests
 # ============================================================================