Spaces:
Sleeping
Sleeping
| """ | |
| Unit tests for configuration data models. | |
| Tests verify that EnvConfig and RewardConfig dataclasses correctly initialize | |
| with default values and accept custom parameters. | |
| """ | |
| import pytest | |
| from env.config import EnvConfig, RewardConfig | |
| class TestEnvConfig: | |
| """Test suite for EnvConfig dataclass.""" | |
| def test_default_values(self): | |
| """Test that EnvConfig initializes with correct default values.""" | |
| config = EnvConfig() | |
| assert config.max_steps == 100 | |
| assert config.initial_resources == 2 | |
| assert config.base_request_rate == 50 | |
| assert config.request_rate_std == 10.0 | |
| assert config.cpu_per_request == 0.5 | |
| assert config.memory_per_request == 0.45 | |
| assert config.resource_capacity == 30.0 | |
| assert config.termination_threshold == 95.0 | |
| def test_custom_parameters(self): | |
| """Test that EnvConfig accepts custom parameter values.""" | |
| config = EnvConfig( | |
| max_steps=200, | |
| initial_resources=5, | |
| base_request_rate=100, | |
| request_rate_std=15.0, | |
| cpu_per_request=0.7, | |
| memory_per_request=0.4, | |
| resource_capacity=50.0, | |
| termination_threshold=90.0 | |
| ) | |
| assert config.max_steps == 200 | |
| assert config.initial_resources == 5 | |
| assert config.base_request_rate == 100 | |
| assert config.request_rate_std == 15.0 | |
| assert config.cpu_per_request == 0.7 | |
| assert config.memory_per_request == 0.4 | |
| assert config.resource_capacity == 50.0 | |
| assert config.termination_threshold == 90.0 | |
| def test_partial_custom_parameters(self): | |
| """Test that EnvConfig allows partial parameter override.""" | |
| config = EnvConfig(max_steps=150, initial_resources=4) | |
| assert config.max_steps == 150 | |
| assert config.initial_resources == 4 | |
| # Verify other parameters use defaults | |
| assert config.base_request_rate == 50 | |
| assert config.request_rate_std == 10.0 | |
| class TestRewardConfig: | |
| """Test suite for RewardConfig dataclass.""" | |
| def test_default_values(self): | |
| """Test that RewardConfig initializes with correct default values.""" | |
| config = RewardConfig() | |
| assert config.target_cpu_min == 40.0 | |
| assert config.target_cpu_max == 70.0 | |
| assert config.target_memory_min == 40.0 | |
| assert config.target_memory_max == 70.0 | |
| assert config.resource_cost_weight == 0.1 | |
| assert config.over_provision_penalty == -0.5 | |
| assert config.under_provision_penalty == -1.0 | |
| def test_custom_parameters(self): | |
| """Test that RewardConfig accepts custom parameter values.""" | |
| config = RewardConfig( | |
| target_cpu_min=30.0, | |
| target_cpu_max=80.0, | |
| target_memory_min=35.0, | |
| target_memory_max=75.0, | |
| resource_cost_weight=0.2, | |
| over_provision_penalty=-0.8, | |
| under_provision_penalty=-1.5 | |
| ) | |
| assert config.target_cpu_min == 30.0 | |
| assert config.target_cpu_max == 80.0 | |
| assert config.target_memory_min == 35.0 | |
| assert config.target_memory_max == 75.0 | |
| assert config.resource_cost_weight == 0.2 | |
| assert config.over_provision_penalty == -0.8 | |
| assert config.under_provision_penalty == -1.5 | |
| def test_partial_custom_parameters(self): | |
| """Test that RewardConfig allows partial parameter override.""" | |
| config = RewardConfig( | |
| target_cpu_min=35.0, | |
| target_cpu_max=65.0 | |
| ) | |
| assert config.target_cpu_min == 35.0 | |
| assert config.target_cpu_max == 65.0 | |
| # Verify other parameters use defaults | |
| assert config.target_memory_min == 40.0 | |
| assert config.resource_cost_weight == 0.1 | |