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()

    # Notebook should no longer do hash conversion directly;
    # the builder handles it internally.
    assert "bits_tensor_to_hash_bytes" not in source
    assert "np.packbits" not in source
    assert "scene_graph.objects[_obj_id] = ObjectNode" not in source


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

    assert "SceneGraphBuilder" in source
    assert "SceneGraphBuildConfig" in source
    assert "flatten_room_views" in source
    assert "scene_graph, build_artifacts = builder.build_from_room_views" in source
    assert "scene_graph.objects[_obj_id] = ObjectNode" not in source