hindenburg/tests/test_transcription.py

import unittest
from src.hindenburg_api.transcription import Transcription

class TestTranscription(unittest.TestCase):

    def setUp(self):
        self.transcription = Transcription()

    def test_parse_json_transcription(self):
        json_data = {
            "lines": [
                {
                    "endTime": "00:00:18,900",
                    "speakerDesignation": "Journalist",
                    "startTime": "00:00:02,739",
                    "text": "Det ska ju också klippa sig ner en massa såklart, så är det någon fråga att bara tänka på eller jag förstår inte frågan Men då får vi inte bara börja med att säga vad heter och vad gör en introduktion"
                },
                {
                    "endTime": "00:00:35,060",
                    "speakerDesignation": "Lars Rudstam",
                    "startTime": "00:00:19,440",
                    "text": "Jag heter Lars Utstam och jag är professor på Cornell universitetet jag är fiskeri och akvatisk ekologi är mina ämnen Jag studerar allting från, i princip allt som finns i vatten."
                }
            ]
        }
        self.transcription.parse_json(json_data)
        self.assertEqual(len(self.transcription.lines), 2)

    def test_convert_to_xml(self):
        self.transcription.lines = [
            {
                "endTime": "00:00:18,900",
                "speakerDesignation": "Journalist",
                "startTime": "00:00:02,739",
                "text": "Test text"
            }
        ]
        xml_output = self.transcription.convert_to_xml()
        self.assertIn("<Transcription", xml_output)
        self.assertIn("<w", xml_output)

if __name__ == '__main__':
    unittest.main()