File size: 1,014 Bytes
31ade1f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from pathlib import Path
from train.dataset_build_utils import append_suffix_once, dataset_version_with_suffix, output_dataset_path
def test_append_suffix_once_idempotent():
assert append_suffix_once("proxy_train_phase", "selector_align") == "proxy_train_phase_selector_align"
assert append_suffix_once("proxy_train_phase_selector_align", "selector_align") == "proxy_train_phase_selector_align"
def test_output_dataset_path_keeps_existing_suffix():
expected = Path("/tmp/proxy_train_phase_selector_align.pt")
assert output_dataset_path("/tmp/proxy_train_phase.pt", "selector_align") == expected
assert output_dataset_path(expected, "selector_align") == expected
def test_dataset_version_with_suffix_keeps_existing_suffix():
assert dataset_version_with_suffix("reveal_proxy_v6_phase", "selector_align") == "reveal_proxy_v6_phase_selector_align"
assert dataset_version_with_suffix("reveal_proxy_v6_phase_selector_align", "selector_align") == "reveal_proxy_v6_phase_selector_align"
|