audiblez/test/test_main.py
Claudio Santini 089f956049
Some checks failed
Git clone and run / linux-git-clone-and-run (push) Has been cancelled
Git clone and run / windows-git-clone-and-run (push) Has been cancelled
Installing via pip and running / install-and-run-on-python-3-11 (push) Has been cancelled
Installing via pip and running / install-and-run-on-python-3-12 (push) Has been cancelled
Installing via pip and running / install-and-run-on-python-3-10 (push) Has been cancelled
Installing via pip and running / windows-git-clone-and-run (push) Has been cancelled
v0.4.9
2025-02-24 10:13:58 +01:00

59 lines
2.4 KiB
Python

import os
import subprocess
import unittest
from pathlib import Path
from ebooklib import epub
from audiblez.core import main, find_document_chapters_and_extract_texts
# from phonemizer.backend.espeak.wrapper import EspeakWrapper
# EspeakWrapper.set_library('/opt/homebrew/Cellar/espeak-ng/1.52.0/lib/libespeak-ng.1.dylib')
class MainTest(unittest.TestCase):
def base(self, name, url='', **kwargs):
if not Path(f'{name}.epub').exists():
os.system(f'wget {url} -O {name}.epub')
Path(f'{name}.m4b').unlink(missing_ok=True)
os.system(f'rm {name}_chapter_*.wav')
merged_args = dict(voice='af_sky', pick_manually=False, speed=1.0, max_chapters=1, max_sentences=2)
merged_args.update(kwargs)
main(f'{name}.epub', **merged_args)
m4b_file = Path(f'{name}.m4b')
self.assertTrue(m4b_file.exists())
self.assertTrue(m4b_file.stat().st_size > 1024)
def test_poe(self):
url = 'https://www.gutenberg.org/ebooks/1064.epub.images'
self.base('poe', url)
def test_orwell(self):
url = 'https://archive.org/download/AnimalFarmByGeorgeOrwell/Animal%20Farm%20by%20George%20Orwell.epub'
self.base('orwell', url)
def test_italian_pirandello_and_filename_with_spaces(self):
url = 'https://www.liberliber.eu/mediateca/libri/p/pirandello/cosi_e_se_vi_pare_1925/epub/pirandello_cosi_e_se_vi_pare_1925.epub'
self.base('pirandello e spazio', url, voice='im_nicola')
self.assertTrue(Path('pirandello e spazio.m4b').exists())
def test_italian_manzoni(self):
url = 'https://www.liberliber.eu/mediateca/libri/m/manzoni/i_promessi_sposi/epub/manzoni_i_promessi_sposi.epub'
self.base('manzoni', url, voice='im_nicola')
def test_french_baudelaire(self):
url = 'http://gallica.bnf.fr/ark:/12148/bpt6k70861t.epub'
self.base('baudelaire', url, voice='ff_siwis')
def test_chinese(self):
url = 'https://www.gutenberg.org/ebooks/24225.epub3.images'
self.base('chinese', url, voice='zf_xiaobei')
def test_leigh_and_play_result(self):
book = epub.read_epub('leigh.epub')
document_chapters = find_document_chapters_and_extract_texts(book)
chapters = [c for c in document_chapters if c.get_name() == 'Text/Chap07.xhtml']
self.base('leigh', voice='af_heart', selected_chapters=chapters, max_sentences=5)
subprocess.run(['ffplay', '-nodisp', '-autoexit', 'leigh.m4b'], check=True)