langchain/libs/community/tests/unit_tests/vectorstores/test_hanavector.py

"""Test HanaVector functionality."""

from langchain_community.vectorstores import HanaDB


def test_int_sanitation_with_illegal_value() -> None:
    """Test sanitization of int with illegal value"""
    successful = True
    try:
        HanaDB._sanitize_int("HUGO")
        successful = False
    except ValueError:
        pass

    assert successful


def test_int_sanitation_with_legal_values() -> None:
    """Test sanitization of int with legal values"""
    assert HanaDB._sanitize_int(42) == 42

    assert HanaDB._sanitize_int("21") == 21


def test_int_sanitation_with_negative_values() -> None:
    """Test sanitization of int with legal values"""
    assert HanaDB._sanitize_int(-1) == -1

    assert HanaDB._sanitize_int("-1") == -1


def test_int_sanitation_with_illegal_negative_value() -> None:
    """Test sanitization of int with illegal value"""
    successful = True
    try:
        HanaDB._sanitize_int(-2)
        successful = False
    except ValueError:
        pass

    assert successful


def test_parse_float_array_from_string() -> None:
    array_as_string = "[0.1, 0.2, 0.3]"
    assert HanaDB._parse_float_array_from_string(array_as_string) == [0.1, 0.2, 0.3]