CloudOptRL / tests /test_stochastic_debug.py
gagan-devv's picture
Initial submission
cb922bf
Raw History Blame Contribute Delete
1.19 kB
#!/usr/bin/env python
"""Debug script to test stochastic behavior."""
import numpy as np
from env.environment import CloudResourceEnv
from env.config import EnvConfig
# Test with two different seeds
config = EnvConfig(
initial_resources=3,
base_request_rate=50,
request_rate_std=10.0
)
env1 = CloudResourceEnv(config=config)
env1.rng.seed(42)
obs1_reset = env1.reset()
print(f"Env1 after reset (seed=42): {obs1_reset}")
print(f" Request rate: {env1.request_rate}")
obs1_step, _, _, _ = env1.step(1) # MAINTAIN action
print(f"Env1 after step: {obs1_step}")
print(f" Request rate: {env1.request_rate}")
print()
env2 = CloudResourceEnv(config=config)
env2.rng.seed(99)
obs2_reset = env2.reset()
print(f"Env2 after reset (seed=99): {obs2_reset}")
print(f" Request rate: {env2.request_rate}")
obs2_step, _, _, _ = env2.step(1) # MAINTAIN action
print(f"Env2 after step: {obs2_step}")
print(f" Request rate: {env2.request_rate}")
print()
print(f"Reset observations equal: {np.allclose(obs1_reset, obs2_reset)}")
print(f"Step observations equal: {np.allclose(obs1_step, obs2_step)}")
print(f"Request rates after step equal: {env1.request_rate == env2.request_rate}")