macronizer / tests /test_online_api.py
Urdatorn's picture
api update
069c647
raw
history blame contribute delete
472 Bytes
from gradio_client import Client
import pytest
@pytest.mark.online_api
def test_online_api_returns_plain_output_only():
client = Client("al1808th/macronizer")
result = client.predict(
text="νεανίας ἀάατός ἐστιν καὶ καλός. τὰ παῖδες τὰ καλά",
model_label="mega",
)
assert result == "νεα_νί^ας ἀ^ά_α^τός ἐστι^ν καὶ κα^λός. τὰ^ παῖδες τὰ^ κα_λά_"