def test_selects_from_first_speaker(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
selected = ()
s1.select(selected, 1, 100).AndReturn("hi")
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.select(selected, 1, 100)
moxer.VerifyAll()
def test_selects_from_first_speaker(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
selected = ()
s1.select(selected, 1, 100).AndReturn("hi")
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.select(selected, 1, 100)
moxer.VerifyAll()
def test_ingests_for_multiple_speakers(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
text = "hello"
s1.ingest(text)
s2.ingest(text)
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.ingest(text)
moxer.VerifyAll()
def test_ingests_for_multiple_speakers(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
text = "hello"
s1.ingest(text)
s2.ingest(text)
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.ingest(text)
moxer.VerifyAll()
def test_selects_from_second_speaker(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
selected = ()
# s1.select() returns None, so s2.select() is called
s1.select(selected, 1, 100)
s2.select(selected, 1, 100).AndReturn("hi")
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.select(selected, 1, 100)
moxer.VerifyAll()
def test_selects_from_second_speaker(self):
moxer = mox.Mox()
s1 = moxer.CreateMock(SelectingSpeaker)
s2 = moxer.CreateMock(SelectingSpeaker)
selected = ()
# s1.select() returns None, so s2.select() is called
s1.select(selected, 1, 100)
s2.select(selected, 1, 100).AndReturn("hi")
moxer.ReplayAll()
speaker = HybridWordSpeaker(s1, s2)
speaker.select(selected, 1, 100)
moxer.VerifyAll()
lambda: NxGraphShortestPathSpeaker(),
'nx.s':
lambda: NxGraphSelectingSpeaker(),
'nx-nx.sp-r':
lambda: HybridPhraseSpeaker(NxGraphSpeaker(), NxGraphShortestPathSpeaker(),
RandomSpeaker()),
'nx-r':
lambda: HybridPhraseSpeaker(NxGraphSpeaker(), RandomSpeaker()),
'm1':
lambda: Markov1Speaker(),
'm1-t':
lambda: Markov1TreeSpeaker(),
'm2':
lambda: Markov2Speaker(),
'm2-m1':
lambda: HybridWordSpeaker(Markov2Speaker(), Markov1Speaker()),
'm2-r':
lambda: HybridPhraseSpeaker(Markov2Speaker(), RandomSpeaker()),
'r':
lambda: RandomSpeaker(),
's':
lambda: SentenceSpeaker(sentences_per_ingest=20),
's-r':
lambda: HybridPhraseSpeaker(SentenceSpeaker(sentences_per_ingest=20),
RandomSpeaker()),
't':
lambda: TreeSpeaker()
}
def speaker_aliases():
