mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-12-31 00:57:37 +00:00
5f3c37d287
* chore(lint): enforce and apply PEP 585 type hinting * Run fmt fixes * Fix all typing imports using some regex * Fix all typing written in string in tests * undo changes in the e2e tests * make e2e test use list instead of List * type fixes * remove type:ignore * pylint * Remove typing from Usage example comments * Remove typing from most of comments * try to fix e2e tests on comm PRs * fix * Add tests typing.List in to adjust test compatiplity - test/components/agents/test_state_class.py - test/components/converters/test_output_adapter.py - test/components/joiners/test_list_joiner.py * simplify pyproject * improve relnote --------- Co-authored-by: anakin87 <stefanofiorucci@gmail.com>
165 lines
5.9 KiB
Python
165 lines
5.9 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from typing import Any, Optional, Union
|
|
|
|
from haystack.core.component.types import GreedyVariadic, Variadic
|
|
from haystack.core.super_component.utils import _is_compatible
|
|
|
|
|
|
class TestTypeCompatibility:
|
|
"""
|
|
Test suite for type compatibility checking functionality.
|
|
"""
|
|
|
|
def test_basic_types(self):
|
|
"""Test compatibility of basic Python types."""
|
|
is_compat, common = _is_compatible(str, str)
|
|
assert is_compat and common == str
|
|
|
|
is_compat, common = _is_compatible(int, int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(str, int)
|
|
assert not is_compat and common is None
|
|
|
|
is_compat, common = _is_compatible(float, int)
|
|
assert not is_compat and common is None
|
|
|
|
def test_any_type(self):
|
|
"""Test Any type compatibility."""
|
|
is_compat, common = _is_compatible(int, Any)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(Any, int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(Any, Any)
|
|
assert is_compat and common == Any
|
|
|
|
is_compat, common = _is_compatible(Any, str)
|
|
assert is_compat and common == str
|
|
|
|
is_compat, common = _is_compatible(str, Any)
|
|
assert is_compat and common == str
|
|
|
|
def test_union_types(self):
|
|
"""Test Union type compatibility."""
|
|
is_compat, common = _is_compatible(int, Union[int, str])
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(Union[int, str], int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(Union[int, str], Union[str, int])
|
|
assert is_compat and common == Union[int, str] or common == Union[str, int]
|
|
|
|
is_compat, common = _is_compatible(str, Union[int, str])
|
|
assert is_compat and common == str
|
|
|
|
is_compat, common = _is_compatible(bool, Union[int, str])
|
|
assert not is_compat and common is None
|
|
|
|
is_compat, common = _is_compatible(float, Union[int, str])
|
|
assert not is_compat and common is None
|
|
|
|
def test_variadic_type_compatibility(self):
|
|
"""Test compatibility with Variadic and GreedyVariadic types."""
|
|
# Basic type compatibility
|
|
variadic_int = Variadic[int]
|
|
greedy_int = GreedyVariadic[int]
|
|
|
|
is_compat, common = _is_compatible(variadic_int, int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(int, variadic_int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(greedy_int, int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(int, greedy_int)
|
|
assert is_compat and common == int
|
|
|
|
# List type compatibility
|
|
variadic_list = Variadic[list[int]]
|
|
greedy_list = GreedyVariadic[list[int]]
|
|
|
|
is_compat, common = _is_compatible(variadic_list, list[int])
|
|
assert is_compat and common == list[int]
|
|
|
|
is_compat, common = _is_compatible(list[int], variadic_list)
|
|
assert is_compat and common == list[int]
|
|
|
|
is_compat, common = _is_compatible(greedy_list, list[int])
|
|
assert is_compat and common == list[int]
|
|
|
|
is_compat, common = _is_compatible(list[int], greedy_list)
|
|
assert is_compat and common == list[int]
|
|
|
|
def test_nested_type_unwrapping(self):
|
|
"""Test nested type unwrapping behavior with unwrap_nested parameter."""
|
|
# Test with unwrap_nested=True (default)
|
|
nested_optional = Variadic[list[Optional[int]]]
|
|
|
|
is_compat, common = _is_compatible(nested_optional, list[int])
|
|
assert is_compat and common == list[int]
|
|
|
|
is_compat, common = _is_compatible(list[int], nested_optional)
|
|
assert is_compat and common == list[int]
|
|
|
|
nested_union = Variadic[list[Union[int, None]]]
|
|
|
|
is_compat, common = _is_compatible(nested_union, list[int])
|
|
assert is_compat and common == list[int]
|
|
|
|
is_compat, common = _is_compatible(list[int], nested_union)
|
|
assert is_compat and common == list[int]
|
|
|
|
def test_complex_nested_types(self):
|
|
"""Test complex nested type scenarios."""
|
|
# Multiple levels of nesting
|
|
complex_type = Variadic[list[list[Variadic[int]]]]
|
|
target_type = list[list[int]]
|
|
|
|
# With unwrap_nested=True
|
|
is_compat, common = _is_compatible(complex_type, target_type)
|
|
assert is_compat and common == list[list[int]]
|
|
|
|
is_compat, common = _is_compatible(target_type, complex_type)
|
|
assert is_compat and common == list[list[int]]
|
|
|
|
# With unwrap_nested=False
|
|
is_compat, common = _is_compatible(complex_type, target_type, unwrap_nested=False)
|
|
assert not is_compat and common is None
|
|
|
|
is_compat, common = _is_compatible(target_type, complex_type, unwrap_nested=False)
|
|
assert not is_compat and common is None
|
|
|
|
def test_mixed_variadic_types(self):
|
|
"""Test mixing Variadic and GreedyVariadic with other type constructs."""
|
|
# Variadic with Union
|
|
var_union = Variadic[Union[int, str]]
|
|
|
|
is_compat, common = _is_compatible(var_union, Union[int, str])
|
|
assert is_compat and common == Union[int, str]
|
|
|
|
is_compat, common = _is_compatible(Union[int, str], var_union)
|
|
assert is_compat and common == Union[int, str]
|
|
|
|
# GreedyVariadic with Optional
|
|
greedy_opt = GreedyVariadic[Optional[int]]
|
|
|
|
is_compat, common = _is_compatible(greedy_opt, int)
|
|
assert is_compat and common == int
|
|
|
|
is_compat, common = _is_compatible(int, greedy_opt)
|
|
assert is_compat and common == int
|
|
|
|
# Nested Variadic and GreedyVariadic
|
|
nested_var = Variadic[list[GreedyVariadic[int]]]
|
|
|
|
is_compat, common = _is_compatible(nested_var, list[int])
|
|
assert is_compat and common == list[int]
|