| 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" | |