test_text_cleaners: don't re-invent assertions

pytest does treat assert special and does a better job w.r.t error reporting
This commit is contained in:
Jörg Thalheim 2020-12-10 16:43:07 +01:00
parent ea2c57b0ac
commit cdb81f8c5d
No known key found for this signature in database
GPG Key ID: 003F2096411B5F92
1 changed files with 7 additions and 15 deletions

View File

@ -1,24 +1,16 @@
#!/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")
assert english_cleaners("It's 11:00") == "it's eleven a m"
assert english_cleaners("It's 9:01") == "it's nine oh one a m"
assert english_cleaners("It's 16:00") == "it's four p m"
assert 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")
assert phoneme_cleaners("It's $10.50") == "It's ten dollars fifty cents"
assert phoneme_cleaners("£1.1") == "one pound sterling one penny"
assert phoneme_cleaners("¥1") == "one yen"