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)
