Spaces:
Running
Running
File size: 1,480 Bytes
b6b0c93 | 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 37 38 39 40 41 42 43 | import pytest
from httpx import AsyncClient
@pytest.mark.asyncio
async def test_langid_batch(client: AsyncClient):
"""Verify that language identification works for a list of strings."""
payload = {"src": ["This is English text.", "Ceci est un texte français."]}
response = await client.post("/api/identify-language", json=payload)
assert response.status_code == 200
data = response.json()
# Expect a list of lists of DetectionResult
results = data["results"]
assert isinstance(results, list)
assert len(results) == 2
# First item: English
assert len(results[0]) >= 1
assert results[0][0]["lang"] == "en"
# Second item: French
assert len(results[1]) >= 1
assert results[1][0]["lang"] == "fr"
@pytest.mark.asyncio
async def test_langid_newline_handling(client: AsyncClient):
"""Verify that inputs with newlines are handled gracefully (no 500 error)."""
# Single string with newline
payload_single = {"src": "This text\nhas a newline."}
response = await client.post("/api/identify-language", json=payload_single)
assert response.status_code == 200
data = response.json()
assert data["results"][0]["lang"] == "en"
# Batch with newlines
payload_batch = {"src": ["Line 1\nLine 2", "Another\nline"]}
response = await client.post("/api/identify-language", json=payload_batch)
assert response.status_code == 200
data = response.json()
assert len(data["results"]) == 2
|