PyFlowery/tests/tests.py

import asyncio
import logging
import sys

from pyflowery.models import FloweryAPIConfig
from pyflowery.pyflowery import FloweryAPI

root = logging.getLogger()
root.setLevel(logging.DEBUG)

handler = logging.StreamHandler(sys.stdout)
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
root.addHandler(handler)

api = FloweryAPI(FloweryAPIConfig())

async def test_get_voices():
    """Test the get_voices method"""
    async for voice in api.get_voices():
        if 'en' in voice.language.code:
            api.config.logger.info(voice)

async def test_get_tts():
    """Test the get_tts method"""
    tts = await api.get_tts(text="BLAST HIM!", voice="191c5adc-a092-5eea-b4ff-ce01f66153ae")
    try:
        with open('test.mp3', 'wb') as f:
            f.write(tts)
    except Exception as e: # pylint: disable=broad-except
        api.config.logger.error(e, exc_info=True)
    long_string = 'a' * 2049
    try:
        await api.get_tts(text=long_string)
    except ValueError as e:
        api.config.logger.error(e, exc_info=True)

if __name__ == '__main__':
    api.config.logger.info("testing get_voices")
    asyncio.run(test_get_voices())
    api.config.logger.info("testing get_tts")
    asyncio.run(test_get_tts())