from __future__ import annotations

from pathlib import Path


NOTEBOOK = Path(__file__).resolve().parents[1] / "notebooks" / "verification.py"


def test_verification_notebook_uses_scenegraph_query_api():
    source = NOTEBOOK.read_text()

    assert "query_image_against_scene_graph" in source
    assert "text_labels=text_labels" in source


def test_verification_notebook_uses_hash_codec_for_object_hashes():
    source = NOTEBOOK.read_text()

    assert "bits_tensor_to_hash_bytes" in source
    assert "np.packbits" not in source