CloudOptRL / tests /test_new_case.py
gagan-devv's picture
Initial submission
cb922bf
Raw History Blame Contribute Delete
1.33 kB
#!/usr/bin/env python
"""Test the new failing case."""
import numpy as np
from env.environment import CloudResourceEnv
from env.config import EnvConfig
# New failing case
action = 0 # DECREASE
initial_resources = 2
base_request_rate = 54
seed1 = 0
seed2 = 3
config = EnvConfig(
initial_resources=initial_resources,
base_request_rate=base_request_rate,
request_rate_std=10.0
)
print("=== ENV1 (seed=0) ===")
env1 = CloudResourceEnv(config=config)
env1.rng.seed(seed1)
env1.reset()
print(f"After reset: request_rate={env1.request_rate}, resources={env1.allocated_resources}")
for i in range(10):
obs1, _, done1, info1 = env1.step(action)
print(f"Step {i+1}: request_rate={info1['request_rate']}, resources={info1['allocated_resources']}, obs={obs1}")
if done1:
print(f" Episode terminated at step {i+1}")
break
print()
print("=== ENV2 (seed=3) ===")
env2 = CloudResourceEnv(config=config)
env2.rng.seed(seed2)
env2.reset()
print(f"After reset: request_rate={env2.request_rate}, resources={env2.allocated_resources}")
for i in range(10):
obs2, _, done2, info2 = env2.step(action)
print(f"Step {i+1}: request_rate={info2['request_rate']}, resources={info2['allocated_resources']}, obs={obs2}")
if done2:
print(f" Episode terminated at step {i+1}")
break