93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
import unittest
|
|
import tempfile
|
|
from pathlib import Path
|
|
from chatmastermind.glossary import Glossary
|
|
|
|
|
|
class TestGlossary(unittest.TestCase):
|
|
|
|
def test_from_file_valid_yaml(self) -> None:
|
|
# Prepare a temporary YAML file with valid content
|
|
with tempfile.NamedTemporaryFile('w', delete=False) as yaml_file:
|
|
yaml_file.write("Name: Sample\n"
|
|
"ID: '123'\n"
|
|
"SourceLang: en\n"
|
|
"TargetLang: es\n"
|
|
"Entries:\n"
|
|
" hello: hola\n"
|
|
" goodbye: adiós\n"
|
|
" 'yes': sí\n") # 'yes' is a YAML keyword and therefore quoted
|
|
yaml_file_path = Path(yaml_file.name)
|
|
|
|
glossary = Glossary.from_file(yaml_file_path)
|
|
self.assertEqual(glossary.name, "Sample")
|
|
self.assertEqual(glossary.source_lang, "en")
|
|
self.assertEqual(glossary.target_lang, "es")
|
|
self.assertEqual(glossary.entries, {"hello": "hola", "goodbye": "adiós", "yes": "sí"})
|
|
yaml_file_path.unlink() # Remove the temporary file
|
|
|
|
def test_to_file_writes_yaml(self) -> None:
|
|
# Create glossary instance
|
|
glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={"yes": "oui"})
|
|
|
|
with tempfile.NamedTemporaryFile('w', delete=False) as tmp_file:
|
|
file_path = Path(tmp_file.name)
|
|
glossary.to_file(file_path)
|
|
|
|
with open(file_path, 'r') as file:
|
|
content = file.read()
|
|
|
|
self.assertIn("Name: Test", content)
|
|
self.assertIn("SourceLang: en", content)
|
|
self.assertIn("TargetLang: fr", content)
|
|
self.assertIn("Entries", content)
|
|
# 'yes' is a YAML keyword and therefore quoted
|
|
self.assertIn("'yes': oui", content)
|
|
file_path.unlink() # Remove the temporary file
|
|
|
|
def test_write_read_glossary(self) -> None:
|
|
# Create glossary instance
|
|
# -> use 'yes' in order to test if the YAML quoting is correctly removed when reading the file
|
|
glossary_write = Glossary(name="Test", source_lang="en", target_lang="fr", entries={"yes": "oui"})
|
|
|
|
with tempfile.NamedTemporaryFile('w', delete=False) as tmp_file:
|
|
file_path = Path(tmp_file.name)
|
|
glossary_write.to_file(file_path)
|
|
|
|
# create new instance from glossary file
|
|
glossary_read = Glossary.from_file(file_path)
|
|
self.assertEqual(glossary_write.name, glossary_read.name)
|
|
self.assertEqual(glossary_write.source_lang, glossary_read.source_lang)
|
|
self.assertEqual(glossary_write.target_lang, glossary_read.target_lang)
|
|
self.assertDictEqual(glossary_write.entries, glossary_read.entries)
|
|
|
|
file_path.unlink() # Remove the temporary file
|
|
|
|
def test_import_export_csv(self) -> None:
|
|
glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={})
|
|
|
|
# First export to CSV
|
|
with tempfile.NamedTemporaryFile('w', delete=False) as csvfile:
|
|
csv_file_path = Path(csvfile.name)
|
|
glossary.entries = {"hello": "salut", "goodbye": "au revoir"}
|
|
glossary.export_csv(glossary.entries, csv_file_path)
|
|
|
|
# Now import CSV
|
|
glossary.import_csv(csv_file_path)
|
|
self.assertEqual(glossary.entries, {"hello": "salut", "goodbye": "au revoir"})
|
|
csv_file_path.unlink() # Remove the temporary file
|
|
|
|
def test_import_export_tsv(self) -> None:
|
|
glossary = Glossary(name="Test", source_lang="en", target_lang="fr", entries={})
|
|
|
|
# First export to TSV
|
|
with tempfile.NamedTemporaryFile('w', delete=False) as tsvfile:
|
|
tsv_file_path = Path(tsvfile.name)
|
|
glossary.entries = {"hello": "salut", "goodbye": "au revoir"}
|
|
glossary.export_tsv(glossary.entries, tsv_file_path)
|
|
|
|
# Now import TSV
|
|
glossary.import_tsv(tsv_file_path)
|
|
self.assertEqual(glossary.entries, {"hello": "salut", "goodbye": "au revoir"})
|
|
tsv_file_path.unlink() # Remove the temporary file
|