Spaces:
Sleeping
Sleeping
File size: 3,866 Bytes
cb922bf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | """
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
|