mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-06 08:30:31 +00:00
85258f0654
* Fix types in test_run.py * Get test_run.py to pass fmt-check * Add test_run to mypy checks * Update test folder to pass ruff linting * Fix merge * Fix HF tests * Fix hf test * Try to fix tests * Another attempt * minor fix * fix SentenceTransformersDiversityRanker * skip integrations tests due to model unavailable on HF inference --------- Co-authored-by: anakin87 <stefanofiorucci@gmail.com>
35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
|
|
from haystack.components.converters.utils import normalize_metadata
|
|
|
|
|
|
def test_normalize_metadata_None():
|
|
assert normalize_metadata(None, sources_count=1) == [{}]
|
|
assert normalize_metadata(None, sources_count=3) == [{}, {}, {}]
|
|
|
|
|
|
def test_normalize_metadata_single_dict():
|
|
assert normalize_metadata({"a": 1}, sources_count=1) == [{"a": 1}]
|
|
assert normalize_metadata({"a": 1}, sources_count=3) == [{"a": 1}, {"a": 1}, {"a": 1}]
|
|
|
|
|
|
def test_normalize_metadata_list_of_right_size():
|
|
assert normalize_metadata([{"a": 1}], sources_count=1) == [{"a": 1}]
|
|
assert normalize_metadata([{"a": 1}, {"b": 2}, {"c": 3}], sources_count=3) == [{"a": 1}, {"b": 2}, {"c": 3}]
|
|
|
|
|
|
def test_normalize_metadata_list_of_wrong_size():
|
|
with pytest.raises(ValueError, match="The length of the metadata list must match the number of sources."):
|
|
normalize_metadata([{"a": 1}], sources_count=3)
|
|
with pytest.raises(ValueError, match="The length of the metadata list must match the number of sources."):
|
|
assert normalize_metadata([{"a": 1}, {"b": 2}, {"c": 3}], sources_count=1)
|
|
|
|
|
|
def test_normalize_metadata_other_type():
|
|
with pytest.raises(ValueError, match="meta must be either None, a dictionary or a list of dictionaries."):
|
|
normalize_metadata(({"a": 1},), sources_count=1)
|