tests/LLMs/test_LLM_class.py

import pytest
import os
from _utils.LLMs.LLM_class import LLM
from _utils.splitters.Splitter_class import Splitter
from _utils.models.gerar_relatorio import (
    DocumentChunk,
)


class TestSplitters:
    llm = LLM()

    @pytest.mark.asyncio
    async def test_google_gemini_modelo_padrao(self, monkeypatch):
        gemini = self.llm.google_gemini()
        result = gemini.invoke("Qual a capital do Brasil?").content

        assert isinstance(result, str)
        assert len(result) > 0

    @pytest.mark.asyncio
    async def test_google_gemini_modelo_flash_2(self, monkeypatch):
        gemini = self.llm.google_gemini("gemini-2.0-flash")
        result = gemini.invoke("Qual a capital do Brasil?").content

        assert isinstance(result, str)
        assert len(result) > 0

    @pytest.mark.asyncio
    async def test_deepseek_modelo_padrao(self, monkeypatch):
        deepseek = self.llm.deepseek()
        result = deepseek.invoke("Qual a capital do Brasil?").content

        assert isinstance(result, str)
        assert len(result) > 0