def test_multiple_sentences(self):
source = "You are conflicted. Put your weapons away."
sentences = self.tagger.tag(source).split('\n')
actual = ""
expected = "Conflicted, you are. Away put your weapons."
for pos_tagged in sentences:
s = Sentence(pos_tagged)
apply_yodish_grammar(s)
actual += s.render() + ' '
actual = actual.strip()
self.assertEqual(expected, actual)
def test_long_input(self):
source = "One two three. Four five six seven eight nine. Ten eleven twelve thirteen. Fourteen fifteen sixteen seventeen eighteen nineteen twenty"
sentences = self.tagger.tag(source).replace('\n', '').split(')(')
actual = ""
expected = "One two three. Four five six seven eight nine. Ten eleven twelve thirteen. Fourteen fifteen sixteen seventeen eighteen nineteen twenty."
for pos_tagged in sentences:
s = Sentence(pos_tagged)
apply_yodish_grammar(s)
actual += s.render() + ' '
actual = actual.strip()
self.assertEqual(expected, actual)
def translate(source):
t = PartOfSpeechTagger('http://text-processing.com/api/tag/', 'text')
source = expand_contractions(source)
source = t.tag(source)
source = source.replace('\n', '')
logging.warning("Foo: " + source)
sentences = source.split(')(')
translated = ""
for pos_tagged in sentences:
s = Sentence(pos_tagged)
apply_yodish_grammar(s)
translated += s.render() + ' '
return translated.strip()
def setUp(self):
self.sut = Sentence()
self.sut.add_end(Word('you', 'PRP'))
self.sut.add_end(Word('are', 'VBP'))
self.sut.add_end(Word('conflicted', 'VBN'))
def test_so_certain_are_you(self):
sut = Sentence(self.tagger.tag("Are you so certain?"))
apply_yodish_grammar(sut)
self.assertEqual("So certain are you?", sut.render())
def test_size_matters_not(self):
sut = Sentence(self.tagger.tag("Size does not matter."))
apply_yodish_grammar(sut)
self.assertEqual("Size matters not.", sut.render())
def test_my_home_this_is(self):
sut = Sentence(self.tagger.tag("This is my home."))
apply_yodish_grammar(sut)
self.assertEqual("My home this is.", sut.render())
def test_away_put_weapons(self):
sut = Sentence(self.tagger.tag("Put your weapons away."))
apply_yodish_grammar(sut)
self.assertEqual("Away put your weapons.", sut.render())
def test_much_anger(self):
sut = Sentence(self.tagger.tag("I sense much anger in him."))
apply_yodish_grammar(sut)
self.assertEqual("Much anger in him, I sense.", sut.render())
def test_uppercase_i(self):
sut = Sentence(self.tagger.tag("ii i i"))
apply_yodish_grammar(sut)
self.assertEqual("Ii I I.", sut.render())
def test_conflicted_you_are(self):
sut = Sentence(self.tagger.tag("You are conflicted."))
apply_yodish_grammar(sut)
self.assertEqual("Conflicted, you are.", sut.render())
