Function test_symmetry

tests/test_codebook.py:33–37

(d, bits)

@pytest.mark.parametrize("d", [64, 128, 256])
@pytest.mark.parametrize("bits", [2, 3, 4])
def test_symmetry(d, bits):
    cb = LloydMaxCodebook(d, bits)
    assert cb.centroids.sum().abs().item() < 1e-4, (
        f"Centroids should be symmetric: sum={cb.centroids.sum().item()}"
    )


@pytest.mark.parametrize("bits", [1, 2, 3, 4])