Function test_mobius_transform_i32

library/python/hnsw/ut/test.py:213–241

()

def test_mobius_transform_i32():
    EPS = 1e-6
    vectors = np.array(
        [
            [12, 13, -14, 15],
            [1000000000, 1000000000, 1000000000, 1000000000],
            [1, 0, 0, 0],
            [-1, -1, -1, -1]
        ],
        np.int32
    )
    expected_vectors = [
        [0.0163487738, 0.0177111717, -0.0190735695, 0.0204359673],
        [1.25e-10, 1.25e-10, 1.25e-10, 1.25e-10],
        [1, 0, 0, 0],
        [-0.25, -0.25, -0.25, -0.25]
    ]
    vector_bytes = vectors.tobytes()
    pool = Pool.from_bytes(vector_bytes, EVectorComponentType.I32, 4)
    transformed_pool = Pool.from_bytes(bytes(0), EVectorComponentType.Float, 4)
    transformed_pool = transform_mobius(pool)
    for i in range(transformed_pool.get_num_items()):
        for j in range(transformed_pool.dimension):
            assert abs(transformed_pool.get_item(i)[j] - expected_vectors[i][j]) < EPS
            assert pool.get_item(i)[j] == vectors[i][j]
    assert transformed_pool.dimension == 4
    assert transformed_pool.get_num_items() == 4
    assert transformed_pool.dtype == EVectorComponentType.Float
    assert pool.dtype == EVectorComponentType.I32


@pytest.mark.parametrize('pool_params', POOL_PARAMS, ids=POOL_IDS)