From dcf3ddddff15d709f1d802000e0e2ca4614b8669 Mon Sep 17 00:00:00 2001 From: Mayank Jobanputra Date: Tue, 2 May 2023 10:00:22 +0200 Subject: [PATCH] Added deprecation tests for seq2seq generator and RAG Generator (#4782) --- test/nodes/test_generator.py | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/test/nodes/test_generator.py b/test/nodes/test_generator.py index a3fa7646d..dde821fb0 100644 --- a/test/nodes/test_generator.py +++ b/test/nodes/test_generator.py @@ -5,11 +5,32 @@ from unittest.mock import patch import pytest from haystack.schema import Document -from haystack.nodes.answer_generator import Seq2SeqGenerator, OpenAIAnswerGenerator +from haystack.nodes.answer_generator import Seq2SeqGenerator, OpenAIAnswerGenerator, RAGenerator from haystack.pipelines import GenerativeQAPipeline from haystack.nodes import PromptTemplate import logging +from ..conftest import fail_at_version + + +@pytest.mark.unit +@fail_at_version(1, 18) +def test_seq2seq_deprecation(): + with pytest.warns(DeprecationWarning): + try: + Seq2SeqGenerator("non_existing_model/model") + except OSError: + pass + + +@pytest.mark.unit +@fail_at_version(1, 18) +def test_rag_deprecation(): + with pytest.warns(DeprecationWarning): + try: + RAGenerator("non_existing_model/model") + except OSError: + pass @pytest.mark.integration