| from cicd_debug_env.env import CICDDebugEnv |
| from cicd_debug_env.models import Action |
|
|
| def test_env_reset(): |
| env = CICDDebugEnv() |
| obs = env.reset() |
| assert obs.pipeline_yaml != "" |
| assert isinstance(obs.step_blame_scores, dict) |
|
|
| def test_env_step(): |
| env = CICDDebugEnv() |
| env.reset() |
| action = Action("analyze_error", {}, 0.9, "testing") |
| obs, reward, done, info = env.step(action) |
| assert not done |
| assert "reward_breakdown" in info |
| assert len(obs.episode_history) > 0 |
| |
| def test_submit_solution_ends_episode(): |
| env = CICDDebugEnv() |
| env.reset() |
| action = Action("submit_solution", {}, 0.9, "testing") |
| obs, reward, done, info = env.step(action) |
| assert done |
|
|
