ViSS-8B / code /tests /test_source_aligned_sft_release.py
mis2's picture
Release ViSS-8B weights, processors and source code
78ac017 verified
Raw History Blame Contribute Delete
4.86 kB
from pathlib import Path
from data_pipeline.temporal_scope.build_source_aligned_sft_release import (
SourceAlignedSFTError,
_canonical_media_path,
_semantic_identity,
_teacher_used_for_labels,
)
import pytest
def _row(image: str = "media/one.jpg", question: str = "Which one?") -> dict:
return {
"messages": [
{
"role": "user",
"loss": False,
"content": [
{"type": "image", "image": image},
{"type": "text", "text": question},
],
},
{
"role": "assistant",
"loss": True,
"content": [{"type": "text", "text": "A"}],
},
]
}
def test_semantic_identity_binds_pixels_question_and_answer() -> None:
inventory = {"media/one.jpg": {"sha256": "a" * 64}}
baseline = _semantic_identity(_row(), inventory)
assert baseline == _semantic_identity(_row(question=" Which one? "), inventory)
assert baseline != _semantic_identity(_row(question="Which event?"), inventory)
changed = {"media/one.jpg": {"sha256": "b" * 64}}
assert baseline != _semantic_identity(_row(), changed)
def test_canonical_media_path_is_content_addressed() -> None:
digest = "ab" + "1" * 62
assert _canonical_media_path("legacy/frame.JPG", digest) == (
f"media/sha256/ab/{digest}.jpg"
)
def test_teacher_label_provenance_is_strict_and_propagatable() -> None:
assert _teacher_used_for_labels(
{"contracts": {"teacher_used_for_labels": True}}
)
assert not _teacher_used_for_labels(
{"contracts": {"teacher_used_for_labels": False}}
)
with pytest.raises(SourceAlignedSFTError):
_teacher_used_for_labels({"contracts": {}})
def test_legacy_v62di_loader_defaults_narrow_but_supports_committed_full_pool() -> None:
from data_pipeline.temporal_scope.build_source_aligned_sft_release import _load_legacy_v62di_source
with pytest.raises(SourceAlignedSFTError, match="Private legacy adapters"):
_load_legacy_v62di_source(Path("synthetic-unused-source"))
def test_legacy_repcount_normalization_is_hash_pinned_and_field_narrow() -> None:
from data_pipeline.temporal_scope.build_source_aligned_sft_release import _load_legacy_repcount_source
with pytest.raises(SourceAlignedSFTError, match="Private legacy adapters"):
_load_legacy_repcount_source(Path("synthetic-unused-source"))
def test_pinned_legacy_inputs_count_as_sealed_sources() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert "sealed_source_count" in text
assert 'getattr(args, "legacy_v62di_source", None)' in text
assert 'getattr(args, "legacy_repcount_source", None)' in text
def test_hash_bound_parent_selection_allows_one_source_filtering_only() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert 'getattr(args, "include_parent_selection", None) is None' in text
assert "unless a parent selection is supplied" in text
def test_composite_source_stage_filter_is_range_and_manifest_bound() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert "--include-source-stage-release-id" in text
assert "requested source stages are absent" in text
assert "source curriculum stage row count diverged" in text
assert '"included_source_stage_release_ids"' in text
def test_capability_parent_selection_is_hash_bound_and_fail_closed() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert "--include-parent-selection" in text
assert "iclr_video.capability_parent_selection.v0" in text
assert "iclr_video.capability_parent_selection.v1" in text
assert "source_release_membership_verified" in text
assert "selected parent IDs were absent" in text
assert '"parent_selection": parent_selection_record' in text
def test_ordered_curriculum_is_one_run_and_preserves_source_order() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert '"training_runs_required": 1' in text
assert '"source_stage_then_source_row"' in text
assert "if not ordered_curriculum" in text
def test_additive_composition_can_retain_exposures_without_curriculum_order() -> None:
text = Path(
"data_pipeline/temporal_scope/build_source_aligned_sft_release.py"
).read_text()
assert "--retain-semantic-exposures" in text
assert "ordered_curriculum or retain_semantic_exposures" in text
assert '"retain_semantic_exposures": retain_semantic_exposures' in text