added tests to get_language_info

llmtranslate/translator.py

@@ -1,14 +1,11 @@
 import asyncio
-import json
-import re
 from langchain_core.prompts import ChatPromptTemplate
 from pydantic import BaseModel, Field
 from langchain_core.language_models import BaseChatModel
-from llmtranslate.exceptions import MissingAPIKeyError, NoneAPIKeyProvidedError, InvalidModelName
 from llmtranslate.utils.available_languages import get_language_info
 from pydantic import BaseModel
 from llmtranslate.utils.text_splitter import split_text_to_chunks, get_first_n_words
-from abc import ABC, abstractmethod
+from abc import ABC
 
 
 MAX_LENGTH = 1000

tests/test_available_languages.py

@@ -0,0 +1,41 @@
+import pytest
+
+from llmtranslate import get_language_info
+
+
+@pytest.mark.parametrize(
+    "identifier,expected",
+    [
+        # Valid inputs
+        ("English", {
+            "language_name": "English",
+            "ISO_639_1_code": "en",
+            "ISO_639_2_code": "eng",
+            "ISO_639_3_code": "eng",
+        }),
+        ("en", {
+            "language_name": "English",
+            "ISO_639_1_code": "en",
+            "ISO_639_2_code": "eng",
+            "ISO_639_3_code": "eng",
+        }),
+        ("spa", {
+            "language_name": "Spanish",
+            "ISO_639_1_code": "es",
+            "ISO_639_2_code": "spa",
+            "ISO_639_3_code": "spa",
+        }),
+        ("eng", {
+            "language_name": "English",
+            "ISO_639_1_code": "en",
+            "ISO_639_2_code": "eng",
+            "ISO_639_3_code": "eng",
+        }),
+        # Invalid inputs
+        ("xyz", None),
+        ("", None),
+        ("123", None),
+    ],
+)
+def test_get_language_info(identifier, expected):
+    assert get_language_info(identifier) == expected