def test_compute_without_score(self): """test that i can compute and erase the results afterwards""" with warnings.catch_warnings(): warnings.filterwarnings("ignore") with make("rte_case5_example", test=True) as env: stats = EpisodeStatistics(env, "test") stats.compute(nb_scenario=2, max_step=10, pbar=False) # i can access it prods, ids_ = stats.get("prod_p") assert prods.shape == (22, 2), "error on the prods shape" assert ids_.shape == (22, 1), "error on the ids shape" with self.assertRaises(RuntimeError): scores, ids_ = stats.get("scores") # i can clear everything stats.clear_all() assert not os.path.exists(os.path.join(env.get_path_env(), stats.get_name_dir("test")))
def test_compute_with_score(self): """test that i can compute and erase the results afterwards""" with warnings.catch_warnings(): warnings.filterwarnings("ignore") with make("rte_case5_example", test=True) as env: stats = EpisodeStatistics(env, "test") stats.compute(nb_scenario=2, max_step=10, pbar=False, scores_func=L2RPNSandBoxScore) # i can access it scores, ids_ = stats.get(EpisodeStatistics.SCORES) assert scores.shape == (20,), "error on the score shape" assert ids_.shape == (20, 1), "error on the ids shape" scores, ids_ = stats.get("scores") assert scores.shape == (20,), "error on the score shape" assert ids_.shape == (20, 1), "error on the ids shape" # i can clear everything stats.clear_all() assert not os.path.exists(os.path.join(env.get_path_env(), stats.get_name_dir("test")))
def test_compute_erase(self): """test that i can compute and erase the results afterwards""" with warnings.catch_warnings(): warnings.filterwarnings("ignore") with make("rte_case5_example", test=True) as env: stats = EpisodeStatistics(env, "test") stats.compute(nb_scenario=1, max_step=10, pbar=False) # the file have been created assert os.path.exists(os.path.join(env.get_path_env(), stats.get_name_dir("test"))) # i can access it aor_, ids_ = stats.get("a_or") assert aor_.shape == (11, 8) # i can clear the data of individual episode stats.clear_episode_data() assert not os.path.exists(os.path.join(env.get_path_env(), stats.get_name_dir("test"), "00")) # i can clear everything stats.clear_all() assert not os.path.exists(os.path.join(env.get_path_env(), stats.get_name_dir("test")))