from tombfall_env import TombfallEnv


def main():
    env = TombfallEnv(render_mode="ansi")
    observation, info = env.reset(seed=42)
    total_reward = 0.0

    for _ in range(1200):
        action = env.action_space.sample()
        observation, reward, terminated, truncated, info = env.step(action)
        total_reward += reward
        if terminated or truncated:
            break

    print(env.render())
    print(f"total_reward={total_reward:.2f}")
    print(f"info={info}")
    _ = observation


if __name__ == "__main__":
    main()