def test_list_nonexistent_directory(tmp_path: pathlib.Path) -> None: base_path = pathlib.Path("/non-existent-directory") sync_path = tensorboard.get_sync_path(test_util.get_dummy_env()) manager = tensorboard.SharedFSTensorboardManager(str(tmp_path), base_path, sync_path) assert not pathlib.Path(base_path).exists() assert manager.list_tfevents() == []
def test_list_tb_files_nonexistent_directory(tmp_path: pathlib.Path) -> None: env = get_dummy_env() base_path = pathlib.Path("/non-existent-directory") sync_path = get_sync_path(env.det_cluster_id, env.det_experiment_id, env.det_trial_id) manager = SharedFSTensorboardManager(str(tmp_path), base_path, sync_path) assert not pathlib.Path(base_path).exists() assert manager.list_tb_files(0) == []
def test_list_directory(tmp_path: pathlib.Path) -> None: base_path = tensorboard.get_base_path({"base_path": BASE_PATH}, manager=True) sync_path = tensorboard.get_sync_path(test_util.get_dummy_env()) manager = tensorboard.SharedFSTensorboardManager(str(tmp_path), base_path, sync_path) full_event_path = BASE_PATH.joinpath("tensorboard", "events.out.tfevents.example") assert set(manager.list_tfevents()) == {full_event_path}
def test_list_tb_files(tmp_path: pathlib.Path) -> None: env = get_dummy_env() base_path = get_base_path({"base_path": BASE_PATH}, manager=True) sync_path = get_sync_path(env.det_cluster_id, env.det_experiment_id, env.det_trial_id) manager = SharedFSTensorboardManager(str(tmp_path), base_path, sync_path) test_files = [ "79375caf89e9.kernel_stats.pb", "79375caf89e9.memory_profile.json.gz", "events.out.tfevents.example", ] test_filepaths = [BASE_PATH.joinpath("tensorboard", test_file) for test_file in test_files] tb_files = manager.list_tb_files(0) assert set(test_filepaths) == set(tb_files)