def test_remove_omissions_multiple_omissions_no_space_before_terminator(
self):
"""Test with 3 omissions and no space before terminator."""
actual_output = CHATUtteranceCleaner.remove_omissions(
'0but where 0is my 0truck?')
desired_output = 'where my?'
self.assertEqual(actual_output, desired_output)
def test_remove_omissions_null_utterance_no_whitespace(self):
"""Test with null utterance and no whitespace."""
utterance = '0[=! applauses]'
actual_output = CHATUtteranceCleaner.remove_omissions(utterance)
desired_output = '0[=! applauses]'
self.assertEqual(actual_output, desired_output)
def test_remove_omissions_empty_string(self):
"""Test remove_omissions with an empty string."""
actual_output = CHATUtteranceCleaner.remove_omissions('')
desired_output = ''
self.assertEqual(actual_output, desired_output)
def test_remove_omissions_omission_in_brackets(self):
"""Test remove_omissions with omission in square brackets."""
utterance = 'This [* 0is] what ?'
actual_output = CHATUtteranceCleaner.remove_omissions(utterance)
desired_output = 'This [* 0is] what ?'
self.assertEqual(actual_output, desired_output)
def test_remove_omissions_multiple_omissions_space_before_terminator(self):
"""Test remove_omissions with an utt containing 3 omissions."""
actual_output = CHATUtteranceCleaner.remove_omissions(
'0but where 0is my 0truck ?')
desired_output = 'where my ?'
self.assertEqual(actual_output, desired_output)
def test_remove_omissions_single_omission(self):
"""Test remove_omissions with an utt containing 1 omission."""
actual_output = CHATUtteranceCleaner.remove_omissions(
'where 0is my truck?')
desired_output = 'where my truck?'
self.assertEqual(actual_output, desired_output)