import unittest

from audio.audioanalyser_diarization import AudioAnalyserDiarization
from audio.audioanalyser_anglais import AudioAnalyserAnglais


class MyTestCase(unittest.TestCase):
    def test_stt_en(self):
        with open("../testsdata/audio/en.wav", "rb") as filecontent:
            #print(filecontent.read())
            res = AudioAnalyserAnglais.stt(filecontent.read())
            self.assertEqual(res, True)  # add assertion here

    def test_diarization(self):
        with open("../testsdata/audio/en.wav", "rb") as filecontent:
            #print(filecontent.read())
            res = AudioAnalyserDiarization.stt(filecontent.read())
            self.assertEqual(res, True)  # add assertion here