Spaces:
Running
Running
File size: 1,054 Bytes
dc376b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
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
|