From da601561740890dc4dc6248e8fd0eaf3a91e0b55 Mon Sep 17 00:00:00 2001 From: "David S. Batista" Date: Mon, 26 May 2025 17:22:51 +0100 Subject: [PATCH] chore: removing unused imports from tests (#9446) --- licenserc.toml | 24 ++++++++++++------- test/components/agents/test_state_class.py | 5 +++- test/components/audio/test_whisper_local.py | 1 + test/components/audio/test_whisper_remote.py | 1 + .../builders/test_answer_builder.py | 1 + .../builders/test_prompt_builder.py | 1 + .../caching/test_url_cache_checker.py | 1 + .../test_document_language_classifier.py | 1 + .../connectors/test_openapi_service.py | 1 + .../test_azure_ocr_doc_converter.py | 2 ++ .../converters/test_csv_to_document.py | 1 + .../converters/test_docx_file_to_document.py | 4 ++++ .../converters/test_html_to_document.py | 1 + .../converters/test_markdown_to_document.py | 1 + .../converters/test_openapi_functions.py | 1 + .../converters/test_pdfminer_to_document.py | 1 + .../converters/test_pptx_to_document.py | 1 + .../converters/test_tika_doc_converter.py | 1 + test/components/converters/test_utils.py | 1 + .../embedders/test_azure_document_embedder.py | 1 + .../embedders/test_azure_text_embedder.py | 1 + ...test_hugging_face_api_document_embedder.py | 1 + .../test_hugging_face_api_text_embedder.py | 1 + .../test_openai_document_embedder.py | 1 + .../embedders/test_openai_text_embedder.py | 1 + ...sentence_transformers_document_embedder.py | 1 + ...sentence_transformers_embedding_backend.py | 1 + ...est_sentence_transformers_text_embedder.py | 1 + .../embedders/types/test_protocol.py | 4 ++++ .../evaluators/test_answer_exact_match.py | 1 + .../test_context_relevance_evaluator.py | 1 + .../evaluators/test_document_map.py | 1 + .../evaluators/test_document_mrr.py | 1 + .../evaluators/test_document_ndcg.py | 1 + .../evaluators/test_document_recall.py | 1 + .../evaluators/test_faithfulness_evaluator.py | 1 + .../evaluators/test_llm_evaluator.py | 3 +-- .../evaluators/test_sas_evaluator.py | 1 + .../extractors/test_llm_metadata_extractor.py | 5 +++- .../fetchers/test_link_content_fetcher.py | 1 + test/components/generators/chat/test_azure.py | 1 + .../generators/chat/test_hugging_face_api.py | 1 + .../components/generators/chat/test_openai.py | 1 + .../generators/chat/test_openai_async.py | 1 + test/components/generators/conftest.py | 1 + test/components/generators/test_azure.py | 1 + .../generators/test_hugging_face_api.py | 1 + .../test_hugging_face_local_generator.py | 2 ++ test/components/generators/test_openai.py | 1 + test/components/joiners/test_answer_joiner.py | 1 + test/components/joiners/test_branch_joiner.py | 1 + .../joiners/test_document_joiner.py | 1 + .../test_csv_document_splitter.py | 2 +- .../preprocessors/test_document_cleaner.py | 1 + .../preprocessors/test_document_splitter.py | 1 + .../preprocessors/test_recursive_splitter.py | 4 ++++ .../preprocessors/test_sentence_tokenizer.py | 4 ++++ .../preprocessors/test_text_cleaner.py | 1 + .../rankers/test_lost_in_the_middle.py | 1 + test/components/rankers/test_metafield.py | 1 + .../test_sentence_transformers_diversity.py | 6 ++--- .../test_sentence_transformers_similarity.py | 1 + .../rankers/test_transformers_similarity.py | 1 + test/components/readers/test_extractive.py | 1 + .../retrievers/test_auto_merging_retriever.py | 4 ++++ .../retrievers/test_filter_retriever.py | 1 + .../test_in_memory_bm25_retriever.py | 1 + .../test_in_memory_embedding_retriever.py | 1 + .../test_sentence_window_retriever.py | 4 ++++ .../routers/test_conditional_router.py | 1 + test/components/routers/test_file_router.py | 1 + .../routers/test_metadata_router.py | 1 + .../routers/test_text_language_router.py | 1 + .../routers/test_transformers_text_router.py | 1 + .../routers/test_zero_shot_text_router.py | 1 + test/components/samplers/test_top_p.py | 1 + test/components/tools/test_tool_invoker.py | 4 ++++ .../components/validators/test_json_schema.py | 1 + test/components/websearch/test_searchapi.py | 1 + test/components/websearch/test_serperdev.py | 1 - .../writers/test_document_writer.py | 1 + .../test_component_signature_validation.py | 4 ++++ test/core/pipeline/features/conftest.py | 4 ++++ test/core/pipeline/features/test_run.py | 4 ++++ test/core/pipeline/test_async_pipeline.py | 4 ++++ test/core/pipeline/test_draw.py | 1 + test/core/pipeline/test_tracing.py | 1 + .../pipeline/test_validation_pipeline_io.py | 3 ++- .../core/sample_components/test_accumulate.py | 1 + test/core/sample_components/test_fstring.py | 1 + .../super_component/test_super_component.py | 5 ++-- test/core/super_component/test_utils.py | 8 +++++-- test/core/test_importing.py | 1 + test/core/test_serialization.py | 1 + test/dataclasses/test_byte_stream.py | 1 + test/dataclasses/test_chat_message.py | 1 + test/dataclasses/test_document.py | 1 + test/dataclasses/test_sparse_embedding.py | 1 + test/dataclasses/test_state.py | 4 ++++ test/dataclasses/test_streaming_chunk.py | 1 + test/marshal/test_yaml.py | 1 + test/test_logging.py | 1 + test/test_telemetry.py | 1 + test/testing/test_factory.py | 1 + test/tools/test_from_function.py | 4 ++++ test/tools/test_toolset.py | 1 - test/tracing/test_datadog.py | 1 + test/tracing/test_opentelemetry.py | 1 + test/tracing/test_tracer.py | 3 ++- test/tracing/test_utils.py | 1 + test/tracing/utils.py | 1 + test/utils/test_auth.py | 1 + test/utils/test_base_serialization.py | 1 + test/utils/test_device.py | 1 + test/utils/test_filters.py | 1 + test/utils/test_hf.py | 1 + test/utils/test_type_serialization.py | 1 + test/utils/test_url_validation.py | 1 + 118 files changed, 187 insertions(+), 26 deletions(-) diff --git a/licenserc.toml b/licenserc.toml index eb0e81005..9c51077cc 100644 --- a/licenserc.toml +++ b/licenserc.toml @@ -1,21 +1,27 @@ headerPath = "license-header.txt" excludes = [ + "*.csv", + "*.feature", + "*.html", + "*.ini", + "*.jinja2", + "*.md", + "*.mp3", + "*.msg", + "*.toml", + "*.txt", + "*.typed", + "*.wav", + "*.yaml", + "*.yml", ".github", + "CITATION.cff", "docker", "docs", "examples", "proposals", "releasenotes", - "test", - "CITATION.cff", - "*.ini", - "*.jinja2", - "*.md", - "*.toml", - "*.txt", - "*.yaml", - "*.wav", ] [properties] diff --git a/test/components/agents/test_state_class.py b/test/components/agents/test_state_class.py index 583d83c6f..43d925dea 100644 --- a/test/components/agents/test_state_class.py +++ b/test/components/agents/test_state_class.py @@ -1,5 +1,8 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest -from typing import List, Dict from dataclasses import dataclass from haystack.dataclasses import ChatMessage diff --git a/test/components/audio/test_whisper_local.py b/test/components/audio/test_whisper_local.py index 8ba6deab6..958045921 100644 --- a/test/components/audio/test_whisper_local.py +++ b/test/components/audio/test_whisper_local.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import sys from pathlib import Path from unittest.mock import patch, MagicMock diff --git a/test/components/audio/test_whisper_remote.py b/test/components/audio/test_whisper_remote.py index 16750f36a..b6d20d9c9 100644 --- a/test/components/audio/test_whisper_remote.py +++ b/test/components/audio/test_whisper_remote.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import pytest diff --git a/test/components/builders/test_answer_builder.py b/test/components/builders/test_answer_builder.py index 47794548e..199647bcc 100644 --- a/test/components/builders/test_answer_builder.py +++ b/test/components/builders/test_answer_builder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest diff --git a/test/components/builders/test_prompt_builder.py b/test/components/builders/test_prompt_builder.py index 88b33a97e..23b874428 100644 --- a/test/components/builders/test_prompt_builder.py +++ b/test/components/builders/test_prompt_builder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Any, Dict, List, Optional from unittest.mock import patch diff --git a/test/components/caching/test_url_cache_checker.py b/test/components/caching/test_url_cache_checker.py index 60b72ff89..706df496a 100644 --- a/test/components/caching/test_url_cache_checker.py +++ b/test/components/caching/test_url_cache_checker.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document, DeserializationError diff --git a/test/components/classifiers/test_document_language_classifier.py b/test/components/classifiers/test_document_language_classifier.py index 9204b3088..eb0806e97 100644 --- a/test/components/classifiers/test_document_language_classifier.py +++ b/test/components/classifiers/test_document_language_classifier.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest diff --git a/test/components/connectors/test_openapi_service.py b/test/components/connectors/test_openapi_service.py index 281534eef..164113fc2 100644 --- a/test/components/connectors/test_openapi_service.py +++ b/test/components/connectors/test_openapi_service.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import json import os from typing import Any, Dict, List diff --git a/test/components/converters/test_azure_ocr_doc_converter.py b/test/components/converters/test_azure_ocr_doc_converter.py index 38e7d1fa5..907797ce5 100644 --- a/test/components/converters/test_azure_ocr_doc_converter.py +++ b/test/components/converters/test_azure_ocr_doc_converter.py @@ -1,7 +1,9 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + # ruff: noqa: E501 + import json import os import os.path diff --git a/test/components/converters/test_csv_to_document.py b/test/components/converters/test_csv_to_document.py index 1b4a273ec..2be910240 100644 --- a/test/components/converters/test_csv_to_document.py +++ b/test/components/converters/test_csv_to_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import os diff --git a/test/components/converters/test_docx_file_to_document.py b/test/components/converters/test_docx_file_to_document.py index 5130ee5ae..8aaf8d4aa 100644 --- a/test/components/converters/test_docx_file_to_document.py +++ b/test/components/converters/test_docx_file_to_document.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import json import os import logging diff --git a/test/components/converters/test_html_to_document.py b/test/components/converters/test_html_to_document.py index 7476b7ecd..c142a969d 100644 --- a/test/components/converters/test_html_to_document.py +++ b/test/components/converters/test_html_to_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging from pathlib import Path diff --git a/test/components/converters/test_markdown_to_document.py b/test/components/converters/test_markdown_to_document.py index 6aa7cadd0..c855382f5 100644 --- a/test/components/converters/test_markdown_to_document.py +++ b/test/components/converters/test_markdown_to_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging from unittest.mock import patch diff --git a/test/components/converters/test_openapi_functions.py b/test/components/converters/test_openapi_functions.py index e15405903..2bc3776a0 100644 --- a/test/components/converters/test_openapi_functions.py +++ b/test/components/converters/test_openapi_functions.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import json import sys import tempfile diff --git a/test/components/converters/test_pdfminer_to_document.py b/test/components/converters/test_pdfminer_to_document.py index 00d751efa..f9a07ea03 100644 --- a/test/components/converters/test_pdfminer_to_document.py +++ b/test/components/converters/test_pdfminer_to_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging from unittest.mock import patch diff --git a/test/components/converters/test_pptx_to_document.py b/test/components/converters/test_pptx_to_document.py index 595e1a123..2d5fad1fe 100644 --- a/test/components/converters/test_pptx_to_document.py +++ b/test/components/converters/test_pptx_to_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import os diff --git a/test/components/converters/test_tika_doc_converter.py b/test/components/converters/test_tika_doc_converter.py index cab23f62c..b240edf40 100644 --- a/test/components/converters/test_tika_doc_converter.py +++ b/test/components/converters/test_tika_doc_converter.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch import pytest diff --git a/test/components/converters/test_utils.py b/test/components/converters/test_utils.py index 296533491..e138f0fce 100644 --- a/test/components/converters/test_utils.py +++ b/test/components/converters/test_utils.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.components.converters.utils import normalize_metadata diff --git a/test/components/embedders/test_azure_document_embedder.py b/test/components/embedders/test_azure_document_embedder.py index 09f803605..528363bef 100644 --- a/test/components/embedders/test_azure_document_embedder.py +++ b/test/components/embedders/test_azure_document_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from openai import APIError diff --git a/test/components/embedders/test_azure_text_embedder.py b/test/components/embedders/test_azure_text_embedder.py index c6474ff02..302193ff4 100644 --- a/test/components/embedders/test_azure_text_embedder.py +++ b/test/components/embedders/test_azure_text_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import pytest diff --git a/test/components/embedders/test_hugging_face_api_document_embedder.py b/test/components/embedders/test_hugging_face_api_document_embedder.py index b5c165cf4..85b8c16a9 100644 --- a/test/components/embedders/test_hugging_face_api_document_embedder.py +++ b/test/components/embedders/test_hugging_face_api_document_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from unittest.mock import MagicMock, patch diff --git a/test/components/embedders/test_hugging_face_api_text_embedder.py b/test/components/embedders/test_hugging_face_api_text_embedder.py index 8d3db192e..bd24e1c0a 100644 --- a/test/components/embedders/test_hugging_face_api_text_embedder.py +++ b/test/components/embedders/test_hugging_face_api_text_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from unittest.mock import MagicMock, patch diff --git a/test/components/embedders/test_openai_document_embedder.py b/test/components/embedders/test_openai_document_embedder.py index f9f15a458..c9bab1873 100644 --- a/test/components/embedders/test_openai_document_embedder.py +++ b/test/components/embedders/test_openai_document_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import random from typing import List diff --git a/test/components/embedders/test_openai_text_embedder.py b/test/components/embedders/test_openai_text_embedder.py index 0a7c571ae..a5476caf2 100644 --- a/test/components/embedders/test_openai_text_embedder.py +++ b/test/components/embedders/test_openai_text_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import pytest diff --git a/test/components/embedders/test_sentence_transformers_document_embedder.py b/test/components/embedders/test_sentence_transformers_document_embedder.py index f2c13796c..4bfe3aaa0 100644 --- a/test/components/embedders/test_sentence_transformers_document_embedder.py +++ b/test/components/embedders/test_sentence_transformers_document_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import random from unittest.mock import MagicMock, patch diff --git a/test/components/embedders/test_sentence_transformers_embedding_backend.py b/test/components/embedders/test_sentence_transformers_embedding_backend.py index be097920a..3de0f0d9c 100644 --- a/test/components/embedders/test_sentence_transformers_embedding_backend.py +++ b/test/components/embedders/test_sentence_transformers_embedding_backend.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch import pytest diff --git a/test/components/embedders/test_sentence_transformers_text_embedder.py b/test/components/embedders/test_sentence_transformers_text_embedder.py index 5f76fd0f5..ea85765af 100644 --- a/test/components/embedders/test_sentence_transformers_text_embedder.py +++ b/test/components/embedders/test_sentence_transformers_text_embedder.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import random from unittest.mock import MagicMock, patch diff --git a/test/components/embedders/types/test_protocol.py b/test/components/embedders/types/test_protocol.py index 923d9a766..8aa93f1d7 100644 --- a/test/components/embedders/types/test_protocol.py +++ b/test/components/embedders/types/test_protocol.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import inspect from typing import Any, Dict diff --git a/test/components/evaluators/test_answer_exact_match.py b/test/components/evaluators/test_answer_exact_match.py index 0c123f0d3..581ef9742 100644 --- a/test/components/evaluators/test_answer_exact_match.py +++ b/test/components/evaluators/test_answer_exact_match.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.components.evaluators import AnswerExactMatchEvaluator diff --git a/test/components/evaluators/test_context_relevance_evaluator.py b/test/components/evaluators/test_context_relevance_evaluator.py index 2d1021f67..804e18ad6 100644 --- a/test/components/evaluators/test_context_relevance_evaluator.py +++ b/test/components/evaluators/test_context_relevance_evaluator.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from typing import List diff --git a/test/components/evaluators/test_document_map.py b/test/components/evaluators/test_document_map.py index 7c7b26c08..4c3109c81 100644 --- a/test/components/evaluators/test_document_map.py +++ b/test/components/evaluators/test_document_map.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/components/evaluators/test_document_mrr.py b/test/components/evaluators/test_document_mrr.py index 88700eed0..25a03b3b9 100644 --- a/test/components/evaluators/test_document_mrr.py +++ b/test/components/evaluators/test_document_mrr.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/components/evaluators/test_document_ndcg.py b/test/components/evaluators/test_document_ndcg.py index 3924855f7..a6be7ce24 100644 --- a/test/components/evaluators/test_document_ndcg.py +++ b/test/components/evaluators/test_document_ndcg.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/components/evaluators/test_document_recall.py b/test/components/evaluators/test_document_recall.py index b7010a793..e2037af6d 100644 --- a/test/components/evaluators/test_document_recall.py +++ b/test/components/evaluators/test_document_recall.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.components.evaluators.document_recall import DocumentRecallEvaluator, RecallMode diff --git a/test/components/evaluators/test_faithfulness_evaluator.py b/test/components/evaluators/test_faithfulness_evaluator.py index 197b727b2..1eb82b475 100644 --- a/test/components/evaluators/test_faithfulness_evaluator.py +++ b/test/components/evaluators/test_faithfulness_evaluator.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import math from typing import List diff --git a/test/components/evaluators/test_llm_evaluator.py b/test/components/evaluators/test_llm_evaluator.py index d7565cd52..d21770101 100644 --- a/test/components/evaluators/test_llm_evaluator.py +++ b/test/components/evaluators/test_llm_evaluator.py @@ -1,14 +1,13 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 -import os + from typing import List import pytest from haystack import Pipeline from haystack.components.evaluators import LLMEvaluator -from haystack.utils.auth import Secret from haystack.dataclasses.chat_message import ChatMessage from haystack.components.generators.chat.openai import OpenAIChatGenerator diff --git a/test/components/evaluators/test_sas_evaluator.py b/test/components/evaluators/test_sas_evaluator.py index f8c5f90c7..ba228e02c 100644 --- a/test/components/evaluators/test_sas_evaluator.py +++ b/test/components/evaluators/test_sas_evaluator.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.components.evaluators.sas_evaluator import SASEvaluator diff --git a/test/components/extractors/test_llm_metadata_extractor.py b/test/components/extractors/test_llm_metadata_extractor.py index cb3ac7525..31417efeb 100644 --- a/test/components/extractors/test_llm_metadata_extractor.py +++ b/test/components/extractors/test_llm_metadata_extractor.py @@ -1,9 +1,12 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import os import pytest from unittest.mock import Mock from haystack import Document, Pipeline -from haystack.components.builders import PromptBuilder from haystack.components.writers import DocumentWriter from haystack.dataclasses import ChatMessage from haystack.document_stores.in_memory import InMemoryDocumentStore diff --git a/test/components/fetchers/test_link_content_fetcher.py b/test/components/fetchers/test_link_content_fetcher.py index 447f05b86..ea2cb14a4 100644 --- a/test/components/fetchers/test_link_content_fetcher.py +++ b/test/components/fetchers/test_link_content_fetcher.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch, Mock import pytest diff --git a/test/components/generators/chat/test_azure.py b/test/components/generators/chat/test_azure.py index e5afcce04..7005c5439 100644 --- a/test/components/generators/chat/test_azure.py +++ b/test/components/generators/chat/test_azure.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from typing import Any, Dict, List, Optional diff --git a/test/components/generators/chat/test_hugging_face_api.py b/test/components/generators/chat/test_hugging_face_api.py index 06acb3107..0ed1b3eb0 100644 --- a/test/components/generators/chat/test_hugging_face_api.py +++ b/test/components/generators/chat/test_hugging_face_api.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from datetime import datetime import os from typing import Any, Dict diff --git a/test/components/generators/chat/test_openai.py b/test/components/generators/chat/test_openai.py index d393e33a1..7ebdbe209 100644 --- a/test/components/generators/chat/test_openai.py +++ b/test/components/generators/chat/test_openai.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch, MagicMock import pytest diff --git a/test/components/generators/chat/test_openai_async.py b/test/components/generators/chat/test_openai_async.py index 7a709b638..cd738be70 100644 --- a/test/components/generators/chat/test_openai_async.py +++ b/test/components/generators/chat/test_openai_async.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import AsyncMock, patch, MagicMock from openai import AsyncOpenAI, OpenAIError diff --git a/test/components/generators/conftest.py b/test/components/generators/conftest.py index 047c8dd77..0441221d2 100644 --- a/test/components/generators/conftest.py +++ b/test/components/generators/conftest.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from datetime import datetime from typing import Iterator from unittest.mock import MagicMock, patch, AsyncMock diff --git a/test/components/generators/test_azure.py b/test/components/generators/test_azure.py index c6df3b22f..0f8ce6092 100644 --- a/test/components/generators/test_azure.py +++ b/test/components/generators/test_azure.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from haystack import Pipeline diff --git a/test/components/generators/test_hugging_face_api.py b/test/components/generators/test_hugging_face_api.py index cba54321f..b10fdecf5 100644 --- a/test/components/generators/test_hugging_face_api.py +++ b/test/components/generators/test_hugging_face_api.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from unittest.mock import MagicMock, Mock, patch from datetime import datetime diff --git a/test/components/generators/test_hugging_face_local_generator.py b/test/components/generators/test_hugging_face_local_generator.py index 02fda159c..c6985ab22 100644 --- a/test/components/generators/test_hugging_face_local_generator.py +++ b/test/components/generators/test_hugging_face_local_generator.py @@ -1,7 +1,9 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + # pylint: disable=too-many-public-methods + from unittest.mock import Mock, patch import pytest diff --git a/test/components/generators/test_openai.py b/test/components/generators/test_openai.py index f0fe989ef..f33b21507 100644 --- a/test/components/generators/test_openai.py +++ b/test/components/generators/test_openai.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import os from typing import List diff --git a/test/components/joiners/test_answer_joiner.py b/test/components/joiners/test_answer_joiner.py index 52e51791e..2da818cd1 100644 --- a/test/components/joiners/test_answer_joiner.py +++ b/test/components/joiners/test_answer_joiner.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/components/joiners/test_branch_joiner.py b/test/components/joiners/test_branch_joiner.py index 05d30ad6d..8cb2c6b81 100644 --- a/test/components/joiners/test_branch_joiner.py +++ b/test/components/joiners/test_branch_joiner.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.components.joiners import BranchJoiner diff --git a/test/components/joiners/test_document_joiner.py b/test/components/joiners/test_document_joiner.py index 8160fdc48..cf9774a67 100644 --- a/test/components/joiners/test_document_joiner.py +++ b/test/components/joiners/test_document_joiner.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import re diff --git a/test/components/preprocessors/test_csv_document_splitter.py b/test/components/preprocessors/test_csv_document_splitter.py index 6c6b8c691..11675e10c 100644 --- a/test/components/preprocessors/test_csv_document_splitter.py +++ b/test/components/preprocessors/test_csv_document_splitter.py @@ -6,7 +6,7 @@ import pytest import logging from pandas import read_csv from io import StringIO -from haystack import Document, Pipeline +from haystack import Document from haystack.core.serialization import component_from_dict, component_to_dict from haystack.components.preprocessors.csv_document_splitter import CSVDocumentSplitter diff --git a/test/components/preprocessors/test_document_cleaner.py b/test/components/preprocessors/test_document_cleaner.py index 57228253f..41f5b4f10 100644 --- a/test/components/preprocessors/test_document_cleaner.py +++ b/test/components/preprocessors/test_document_cleaner.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest diff --git a/test/components/preprocessors/test_document_splitter.py b/test/components/preprocessors/test_document_splitter.py index 81e0fa2ae..95eae4047 100644 --- a/test/components/preprocessors/test_document_splitter.py +++ b/test/components/preprocessors/test_document_splitter.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import List import re diff --git a/test/components/preprocessors/test_recursive_splitter.py b/test/components/preprocessors/test_recursive_splitter.py index d4924deca..67fa8df4b 100644 --- a/test/components/preprocessors/test_recursive_splitter.py +++ b/test/components/preprocessors/test_recursive_splitter.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import re import pytest diff --git a/test/components/preprocessors/test_sentence_tokenizer.py b/test/components/preprocessors/test_sentence_tokenizer.py index a3b7f8c80..06af48a84 100644 --- a/test/components/preprocessors/test_sentence_tokenizer.py +++ b/test/components/preprocessors/test_sentence_tokenizer.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import time import pytest from unittest.mock import patch diff --git a/test/components/preprocessors/test_text_cleaner.py b/test/components/preprocessors/test_text_cleaner.py index 38cd8ade5..8a83728d0 100644 --- a/test/components/preprocessors/test_text_cleaner.py +++ b/test/components/preprocessors/test_text_cleaner.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from haystack.components.preprocessors import TextCleaner diff --git a/test/components/rankers/test_lost_in_the_middle.py b/test/components/rankers/test_lost_in_the_middle.py index edbe13a89..741b41ea7 100644 --- a/test/components/rankers/test_lost_in_the_middle.py +++ b/test/components/rankers/test_lost_in_the_middle.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document from haystack.components.rankers.lost_in_the_middle import LostInTheMiddleRanker diff --git a/test/components/rankers/test_metafield.py b/test/components/rankers/test_metafield.py index 87366e27c..62e379ffe 100644 --- a/test/components/rankers/test_metafield.py +++ b/test/components/rankers/test_metafield.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest diff --git a/test/components/rankers/test_sentence_transformers_diversity.py b/test/components/rankers/test_sentence_transformers_diversity.py index 74994cdd8..69b08c38c 100644 --- a/test/components/rankers/test_sentence_transformers_diversity.py +++ b/test/components/rankers/test_sentence_transformers_diversity.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import MagicMock, call, patch import pytest @@ -8,10 +9,7 @@ import torch from haystack import Document, Pipeline from haystack.components.rankers import SentenceTransformersDiversityRanker -from haystack.components.rankers.sentence_transformers_diversity import ( - DiversityRankingSimilarity, - DiversityRankingStrategy, -) +from haystack.components.rankers.sentence_transformers_diversity import DiversityRankingSimilarity from haystack.utils import ComponentDevice from haystack.utils.auth import Secret diff --git a/test/components/rankers/test_sentence_transformers_similarity.py b/test/components/rankers/test_sentence_transformers_similarity.py index 97bb251dd..8d0836cc5 100644 --- a/test/components/rankers/test_sentence_transformers_similarity.py +++ b/test/components/rankers/test_sentence_transformers_similarity.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import MagicMock, patch import pytest diff --git a/test/components/rankers/test_transformers_similarity.py b/test/components/rankers/test_transformers_similarity.py index be5164cc4..f6f90af97 100644 --- a/test/components/rankers/test_transformers_similarity.py +++ b/test/components/rankers/test_transformers_similarity.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging from unittest.mock import MagicMock, patch diff --git a/test/components/readers/test_extractive.py b/test/components/readers/test_extractive.py index 9a6eb4bbd..f6487d1e7 100644 --- a/test/components/readers/test_extractive.py +++ b/test/components/readers/test_extractive.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging from math import ceil, exp from typing import List diff --git a/test/components/retrievers/test_auto_merging_retriever.py b/test/components/retrievers/test_auto_merging_retriever.py index eb55c9c40..2d235a2e1 100644 --- a/test/components/retrievers/test_auto_merging_retriever.py +++ b/test/components/retrievers/test_auto_merging_retriever.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document, Pipeline diff --git a/test/components/retrievers/test_filter_retriever.py b/test/components/retrievers/test_filter_retriever.py index bef09e7a7..c6e822e17 100644 --- a/test/components/retrievers/test_filter_retriever.py +++ b/test/components/retrievers/test_filter_retriever.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Dict, Any, List import pytest diff --git a/test/components/retrievers/test_in_memory_bm25_retriever.py b/test/components/retrievers/test_in_memory_bm25_retriever.py index ed1ba1887..bf220c935 100644 --- a/test/components/retrievers/test_in_memory_bm25_retriever.py +++ b/test/components/retrievers/test_in_memory_bm25_retriever.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Dict, Any import pytest diff --git a/test/components/retrievers/test_in_memory_embedding_retriever.py b/test/components/retrievers/test_in_memory_embedding_retriever.py index 366fd17b3..02ad6fbca 100644 --- a/test/components/retrievers/test_in_memory_embedding_retriever.py +++ b/test/components/retrievers/test_in_memory_embedding_retriever.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Dict, Any import pytest diff --git a/test/components/retrievers/test_sentence_window_retriever.py b/test/components/retrievers/test_sentence_window_retriever.py index 04e03befb..035ddd76c 100644 --- a/test/components/retrievers/test_sentence_window_retriever.py +++ b/test/components/retrievers/test_sentence_window_retriever.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import DeserializationError, Document, Pipeline diff --git a/test/components/routers/test_conditional_router.py b/test/components/routers/test_conditional_router.py index df2321f6a..dc26e0676 100644 --- a/test/components/routers/test_conditional_router.py +++ b/test/components/routers/test_conditional_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import copy from typing import List from unittest import mock diff --git a/test/components/routers/test_file_router.py b/test/components/routers/test_file_router.py index 2c1bb51b3..d987b9605 100644 --- a/test/components/routers/test_file_router.py +++ b/test/components/routers/test_file_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import io import sys from unittest.mock import mock_open, patch diff --git a/test/components/routers/test_metadata_router.py b/test/components/routers/test_metadata_router.py index 2b7e96470..140a261aa 100644 --- a/test/components/routers/test_metadata_router.py +++ b/test/components/routers/test_metadata_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/components/routers/test_text_language_router.py b/test/components/routers/test_text_language_router.py index c72e55323..4bda40062 100644 --- a/test/components/routers/test_text_language_router.py +++ b/test/components/routers/test_text_language_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest from _pytest.logging import LogCaptureFixture diff --git a/test/components/routers/test_transformers_text_router.py b/test/components/routers/test_transformers_text_router.py index bb751e9c6..51eeb712e 100644 --- a/test/components/routers/test_transformers_text_router.py +++ b/test/components/routers/test_transformers_text_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch, MagicMock import pytest diff --git a/test/components/routers/test_zero_shot_text_router.py b/test/components/routers/test_zero_shot_text_router.py index 6e74dae65..33f8e6ec2 100644 --- a/test/components/routers/test_zero_shot_text_router.py +++ b/test/components/routers/test_zero_shot_text_router.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import patch import pytest diff --git a/test/components/samplers/test_top_p.py b/test/components/samplers/test_top_p.py index 49317fb83..764e623ba 100644 --- a/test/components/samplers/test_top_p.py +++ b/test/components/samplers/test_top_p.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import random from typing import List diff --git a/test/components/tools/test_tool_invoker.py b/test/components/tools/test_tool_invoker.py index 69fb6c4d5..7e1f696df 100644 --- a/test/components/tools/test_tool_invoker.py +++ b/test/components/tools/test_tool_invoker.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest import json import datetime diff --git a/test/components/validators/test_json_schema.py b/test/components/validators/test_json_schema.py index a407d7c10..96d4aa38b 100644 --- a/test/components/validators/test_json_schema.py +++ b/test/components/validators/test_json_schema.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import json from typing import List diff --git a/test/components/websearch/test_searchapi.py b/test/components/websearch/test_searchapi.py index 3eee31c27..f604cdddc 100644 --- a/test/components/websearch/test_searchapi.py +++ b/test/components/websearch/test_searchapi.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from unittest.mock import Mock, patch diff --git a/test/components/websearch/test_serperdev.py b/test/components/websearch/test_serperdev.py index a2fe33344..8ef1be60a 100644 --- a/test/components/websearch/test_serperdev.py +++ b/test/components/websearch/test_serperdev.py @@ -2,7 +2,6 @@ # # SPDX-License-Identifier: Apache-2.0 -import json import os from unittest.mock import Mock, patch from haystack.utils.auth import Secret diff --git a/test/components/writers/test_document_writer.py b/test/components/writers/test_document_writer.py index 27e13b3f3..bb83e22c6 100644 --- a/test/components/writers/test_document_writer.py +++ b/test/components/writers/test_document_writer.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document, DeserializationError diff --git a/test/core/component/test_component_signature_validation.py b/test/core/component/test_component_signature_validation.py index 05d274f54..b7ec56fd1 100644 --- a/test/core/component/test_component_signature_validation.py +++ b/test/core/component/test_component_signature_validation.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + from typing import Dict, Any, List import pytest from haystack.core.component import component diff --git a/test/core/pipeline/features/conftest.py b/test/core/pipeline/features/conftest.py index fbbc7150c..59e05a338 100644 --- a/test/core/pipeline/features/conftest.py +++ b/test/core/pipeline/features/conftest.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + from dataclasses import dataclass, field from typing import Tuple, List, Dict, Any, Set, Union from pathlib import Path diff --git a/test/core/pipeline/features/test_run.py b/test/core/pipeline/features/test_run.py index 7ee32dbf6..3aa8ad238 100644 --- a/test/core/pipeline/features/test_run.py +++ b/test/core/pipeline/features/test_run.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import json from copy import deepcopy from typing import List, Optional, Dict, Any diff --git a/test/core/pipeline/test_async_pipeline.py b/test/core/pipeline/test_async_pipeline.py index fdb05bfd5..39afa3d6c 100644 --- a/test/core/pipeline/test_async_pipeline.py +++ b/test/core/pipeline/test_async_pipeline.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import asyncio from haystack import AsyncPipeline diff --git a/test/core/pipeline/test_draw.py b/test/core/pipeline/test_draw.py index 05f7249dd..cd80f158a 100644 --- a/test/core/pipeline/test_draw.py +++ b/test/core/pipeline/test_draw.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from unittest.mock import MagicMock, patch import pytest diff --git a/test/core/pipeline/test_tracing.py b/test/core/pipeline/test_tracing.py index a643d11f4..60bc39b33 100644 --- a/test/core/pipeline/test_tracing.py +++ b/test/core/pipeline/test_tracing.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from test.tracing.utils import SpyingSpan, SpyingTracer from typing import Optional from unittest.mock import ANY diff --git a/test/core/pipeline/test_validation_pipeline_io.py b/test/core/pipeline/test_validation_pipeline_io.py index 5ca1c0809..2f2b0a4bf 100644 --- a/test/core/pipeline/test_validation_pipeline_io.py +++ b/test/core/pipeline/test_validation_pipeline_io.py @@ -1,6 +1,7 @@ -# SPDX-FileCopyrightText: 2023-present deepset GmbH +# SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Optional import pytest diff --git a/test/core/sample_components/test_accumulate.py b/test/core/sample_components/test_accumulate.py index 8e31cc63d..f9f3ddfc6 100644 --- a/test/core/sample_components/test_accumulate.py +++ b/test/core/sample_components/test_accumulate.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from haystack.testing.sample_components.accumulate import Accumulate, _default_function diff --git a/test/core/sample_components/test_fstring.py b/test/core/sample_components/test_fstring.py index 0013d48bd..de0bf41f6 100644 --- a/test/core/sample_components/test_fstring.py +++ b/test/core/sample_components/test_fstring.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.testing.sample_components import FString diff --git a/test/core/super_component/test_super_component.py b/test/core/super_component/test_super_component.py index 17a1854b3..5adba29d8 100644 --- a/test/core/super_component/test_super_component.py +++ b/test/core/super_component/test_super_component.py @@ -1,7 +1,8 @@ -# SPDX-FileCopyrightText: 2024-present deepset GmbH +# SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 -from typing import Any, List, Optional, Union + +from typing import Any, List, Union import pytest from haystack import Document, SuperComponent, Pipeline, AsyncPipeline, component, super_component diff --git a/test/core/super_component/test_utils.py b/test/core/super_component/test_utils.py index f6ea94154..dd9491052 100644 --- a/test/core/super_component/test_utils.py +++ b/test/core/super_component/test_utils.py @@ -1,7 +1,11 @@ -from typing import Any, List, Optional, Tuple, Union, cast +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + +from typing import Any, List, Optional, Union from haystack.core.component.types import GreedyVariadic, Variadic -from haystack.core.super_component.utils import _is_compatible, get_args +from haystack.core.super_component.utils import _is_compatible class TestTypeCompatibility: diff --git a/test/core/test_importing.py b/test/core/test_importing.py index 5ea84d25b..c3f0341c2 100644 --- a/test/core/test_importing.py +++ b/test/core/test_importing.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import sys from unittest.mock import patch diff --git a/test/core/test_serialization.py b/test/core/test_serialization.py index 45fbb7c64..2a091227d 100644 --- a/test/core/test_serialization.py +++ b/test/core/test_serialization.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import sys from unittest.mock import Mock diff --git a/test/dataclasses/test_byte_stream.py b/test/dataclasses/test_byte_stream.py index 1858aad83..c3f149a0b 100644 --- a/test/dataclasses/test_byte_stream.py +++ b/test/dataclasses/test_byte_stream.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.dataclasses import ByteStream diff --git a/test/dataclasses/test_chat_message.py b/test/dataclasses/test_chat_message.py index 467307e9b..b7f783410 100644 --- a/test/dataclasses/test_chat_message.py +++ b/test/dataclasses/test_chat_message.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest import json diff --git a/test/dataclasses/test_document.py b/test/dataclasses/test_document.py index 9e705b6e4..61ef5a375 100644 --- a/test/dataclasses/test_document.py +++ b/test/dataclasses/test_document.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/dataclasses/test_sparse_embedding.py b/test/dataclasses/test_sparse_embedding.py index 362bf9477..de90a8447 100644 --- a/test/dataclasses/test_sparse_embedding.py +++ b/test/dataclasses/test_sparse_embedding.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.dataclasses.sparse_embedding import SparseEmbedding diff --git a/test/dataclasses/test_state.py b/test/dataclasses/test_state.py index 1dc42017d..c5aa768d9 100644 --- a/test/dataclasses/test_state.py +++ b/test/dataclasses/test_state.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest from typing import List, Dict diff --git a/test/dataclasses/test_streaming_chunk.py b/test/dataclasses/test_streaming_chunk.py index 656c69e38..6f161d87e 100644 --- a/test/dataclasses/test_streaming_chunk.py +++ b/test/dataclasses/test_streaming_chunk.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.dataclasses import StreamingChunk diff --git a/test/marshal/test_yaml.py b/test/marshal/test_yaml.py index 1c8e55021..d5f03e426 100644 --- a/test/marshal/test_yaml.py +++ b/test/marshal/test_yaml.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.marshal.yaml import YamlMarshaller diff --git a/test/test_logging.py b/test/test_logging.py index 8e2bb0a6a..5d454bbff 100644 --- a/test/test_logging.py +++ b/test/test_logging.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import builtins import json import logging diff --git a/test/test_telemetry.py b/test/test_telemetry.py index 18b88c20f..2ba56e0aa 100644 --- a/test/test_telemetry.py +++ b/test/test_telemetry.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import datetime import logging from unittest.mock import Mock, patch diff --git a/test/testing/test_factory.py b/test/testing/test_factory.py index 8ce979fb7..5aaff43f2 100644 --- a/test/testing/test_factory.py +++ b/test/testing/test_factory.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.dataclasses import Document diff --git a/test/tools/test_from_function.py b/test/tools/test_from_function.py index 478d7d59b..e03c7b54f 100644 --- a/test/tools/test_from_function.py +++ b/test/tools/test_from_function.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: 2022-present deepset GmbH +# +# SPDX-License-Identifier: Apache-2.0 + import pytest from haystack.tools.errors import SchemaGenerationError diff --git a/test/tools/test_toolset.py b/test/tools/test_toolset.py index f8c59ed96..ea21ee45a 100644 --- a/test/tools/test_toolset.py +++ b/test/tools/test_toolset.py @@ -3,7 +3,6 @@ # SPDX-License-Identifier: Apache-2.0 import pytest -import os from haystack import Pipeline from haystack.core.serialization import generate_qualified_class_name from haystack.dataclasses import ChatMessage diff --git a/test/tracing/test_datadog.py b/test/tracing/test_datadog.py index 436ba2682..b8f44aa3f 100644 --- a/test/tracing/test_datadog.py +++ b/test/tracing/test_datadog.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import functools import json from typing import List, Dict diff --git a/test/tracing/test_opentelemetry.py b/test/tracing/test_opentelemetry.py index 1163de0cb..c8961f06f 100644 --- a/test/tracing/test_opentelemetry.py +++ b/test/tracing/test_opentelemetry.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import opentelemetry.trace import pytest from opentelemetry.sdk.trace.export import SimpleSpanProcessor diff --git a/test/tracing/test_tracer.py b/test/tracing/test_tracer.py index 2589bd65e..f27ae66f9 100644 --- a/test/tracing/test_tracer.py +++ b/test/tracing/test_tracer.py @@ -1,8 +1,9 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import sys -from unittest.mock import Mock, patch +from unittest.mock import Mock import ddtrace import opentelemetry.trace diff --git a/test/tracing/test_utils.py b/test/tracing/test_utils.py index c4bd5bb13..348fce8cd 100644 --- a/test/tracing/test_utils.py +++ b/test/tracing/test_utils.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from typing import Any, Union import pytest diff --git a/test/tracing/utils.py b/test/tracing/utils.py index 198fd7089..b8c2a5b1d 100644 --- a/test/tracing/utils.py +++ b/test/tracing/utils.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import contextlib import dataclasses import uuid diff --git a/test/utils/test_auth.py b/test/utils/test_auth.py index 31910c4fa..d37da5a47 100644 --- a/test/utils/test_auth.py +++ b/test/utils/test_auth.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os import pytest diff --git a/test/utils/test_base_serialization.py b/test/utils/test_base_serialization.py index 1534d6b10..a4c7049ce 100644 --- a/test/utils/test_base_serialization.py +++ b/test/utils/test_base_serialization.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from haystack.core.errors import DeserializationError, SerializationError import pytest diff --git a/test/utils/test_device.py b/test/utils/test_device.py index cb58ab8fa..88cd68d5c 100644 --- a/test/utils/test_device.py +++ b/test/utils/test_device.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import os from unittest.mock import patch diff --git a/test/utils/test_filters.py b/test/utils/test_filters.py index c7201bff0..c6c0dff1e 100644 --- a/test/utils/test_filters.py +++ b/test/utils/test_filters.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import pytest from haystack import Document diff --git a/test/utils/test_hf.py b/test/utils/test_hf.py index d75e0b750..8c5859056 100644 --- a/test/utils/test_hf.py +++ b/test/utils/test_hf.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import logging import pytest diff --git a/test/utils/test_type_serialization.py b/test/utils/test_type_serialization.py index 4205783b9..3b6ff117d 100644 --- a/test/utils/test_type_serialization.py +++ b/test/utils/test_type_serialization.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + import sys import typing from typing import Any, List, Dict, Set, Tuple, Union, Optional, FrozenSet, Deque diff --git a/test/utils/test_url_validation.py b/test/utils/test_url_validation.py index 089be16e2..8d1abe954 100644 --- a/test/utils/test_url_validation.py +++ b/test/utils/test_url_validation.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 + from haystack.utils.url_validation import is_valid_http_url