TheRock/rocm-systems/experimental/python/perfxpert/tests/test_fence/test_determinism.py

"""Determinism + size guardrail tests for FenceBuilder."""

from pathlib import Path

import pytest

from perfxpert.fence import FenceBuilder


_SLICES_DIR = (
    Path(__file__).parent.parent.parent
    / "perfxpert" / "fence" / "slices"
)


ROLES = [
    "root",
    "analysis",
    "recommendation",
    "correctness",
    "compute_specialist",
    "memory_specialist",
    "latency_specialist",
]


def test_bit_identical_across_three_calls():
    fb = FenceBuilder()
    for role in ROLES:
        a = fb.build(role)
        b = fb.build(role)
        c = fb.build(role)
        assert a == b == c, f"non-deterministic output for role={role}"


def test_each_slice_under_400_lines():
    for slice_path in sorted(_SLICES_DIR.glob("*.md")):
        lines = slice_path.read_text().splitlines()
        assert len(lines) <= 400, (
            f"slice {slice_path.name} has {len(lines)} lines (limit 400)"
        )


def test_full_fence_under_60kb():
    fb = FenceBuilder()
    for role in ROLES:
        size = len(fb.build(role).encode("utf-8"))
        assert size <= 60 * 1024, f"{role} fence is {size} B (limit 60 KB)"


def test_specialization_is_stable():
    fb = FenceBuilder()
    a = fb.build("recommendation", bottleneck="memory_transfer", gfx_id="gfx942")
    b = fb.build("recommendation", bottleneck="memory_transfer", gfx_id="gfx942")
    assert a == b


def test_different_inputs_yield_different_output():
    fb = FenceBuilder()
    mi300 = fb.build("analysis", gfx_id="gfx942")
    mi350 = fb.build("analysis", gfx_id="gfx950")
    assert mi300 != mi350