def test_chars_no_pinyin(self):
"""Should not generate color output if no available reading."""
note = dict.fromkeys(['Color', 'Reading'], '')
for s in ['没有', '没 有', '没有。']:
fill_color(s, note)
self.assertEqual(note['Color'], '')
def test_mixed_english_chinese(self):
note = dict.fromkeys(['Color', 'Reading'], '')
note['Reading'] = 'Brian de'
fill_color('Brian的', note)
self.assertEqual(
note['Color'],
'<span class="tone5">Brian</span><span class="tone5">的</span>',
)
def test_mismatched_inputs(self):
note = dict.fromkeys(['Color'], '')
note['Reading'] = 'Méiyǒu, shì wǒ dìyīcì lái Shànghǎi lǚyóu.'
fill_color('（没有，）是我第一次来上海旅游。', note)
self.assertEqual(
note['Color'],
('<span class="tone2">没</span>'
'<span class="tone3">有</span>，'
'<span class="tone4">是</span>'
'<span class="tone3">我</span>'
'<span class="tone4">第</span>'
'<span class="tone1">一</span>'
'<span class="tone4">次</span>'
'<span class="tone2">来</span>'
'<span class="tone4">上</span>'
'<span class="tone3">海</span>'
'<span class="tone3">旅</span>'
'<span class="tone2">游</span>。'),
)
def test_ungrouped_chars_grouped_pinyin(self):
note = dict.fromkeys(['Color Hanzi'], '')
note['Pinyin'] = 'Méiyǒu, shì wǒ dìyīcì lái Shànghǎi lǚyóu.'
fill_color('没有，是我第一次来上海旅游。', note)
self.assertEqual(
note['Color Hanzi'],
('<span class="tone2">没</span>'
'<span class="tone3">有</span>，'
'<span class="tone4">是</span>'
'<span class="tone3">我</span>'
'<span class="tone4">第</span>'
'<span class="tone1">一</span>'
'<span class="tone4">次</span>'
'<span class="tone2">来</span>'
'<span class="tone4">上</span>'
'<span class="tone3">海</span>'
'<span class="tone3">旅</span>'
'<span class="tone2">游</span>。'),
)
def test_ungrouped_chars_grouped_pinyin(self):
note = {
'Color': '',
'Reading': 'Méiyǒu, shì wǒ dìyīcì lái Shànghǎi lǚyóu.',
}
fill_color('没有，是我第一次来上海旅游。', note)
self.assertEqual(
note['Color'],
('<span class="tone2">没</span>'
'<span class="tone3">有</span>，'
'<span class="tone4">是</span>'
'<span class="tone3">我</span>'
'<span class="tone4">第</span>'
'<span class="tone1">一</span>'
'<span class="tone4">次</span>'
'<span class="tone2">来</span>'
'<span class="tone4">上</span>'
'<span class="tone3">海</span>'
'<span class="tone3">旅</span>'
'<span class="tone2">游</span>。'),
)
