From 2007b0bcee90f9d4a08d22c8c4759d1cad4ab113 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?J=C3=B6rg=20Thalheim?= Date: Sat, 5 Dec 2020 09:46:51 +0100 Subject: [PATCH] add tests for currency/time expansion --- tests/test_text_cleaners.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 tests/test_text_cleaners.py diff --git a/tests/test_text_cleaners.py b/tests/test_text_cleaners.py new file mode 100644 index 00000000..081e819a --- /dev/null +++ b/tests/test_text_cleaners.py @@ -0,0 +1,24 @@ +#!/usr/bin/env python3 + +from TTS.tts.utils.text.cleaners import english_cleaners, phoneme_cleaners +from typing import Any + + +def assert_equal(actual: Any, expected: Any) -> None: + assert actual == expected, f"\n{actual} \n vs \n{expected}" + + +def test_time() -> None: + assert_equal(english_cleaners("It's 11:00"), "it's eleven a m") + assert_equal(english_cleaners("It's 9:01"), "it's nine oh one a m") + assert_equal(english_cleaners("It's 16:00"), "it's four p m") + assert_equal(english_cleaners("It's 00:00 am"), "it's twelve a m") + + +def test_currency() -> None: + assert_equal(phoneme_cleaners("It's $10.50"), + "It's ten dollars fifty cents") + assert_equal(phoneme_cleaners("£1.1"), + "one pound sterling one penny") + assert_equal(phoneme_cleaners("¥1"), + "one yen")