def test_sample_values(self):
A, B = sentences([u'A', u'B']), sentences([u'Y', u'Z'])
samples = list(_aligned_samples(A, B, [(0, 1), (1, 0)]))
# Note alignments swapped so A -> Z and B-> Y
s0 = SentencePair(Sentence([u'A']), Sentence([u'Z']), aligned=True)
s1 = SentencePair(Sentence([u'B']), Sentence([u'Y']), aligned=True)
self.assertEquals(2, len(samples))
self.assertEquals([s0, s1], samples)
def test_sample_values(self):
A, B = sentences([u'A', u'B']), sentences([u'Y', u'Z'])
samples = list(_aligned_samples(A, B, [(0, 1), (1, 0)]))
# Note alignments swapped so A -> Z and B-> Y
s0 = SentencePair(Sentence([u'A']), Sentence([u'Z']), aligned=True)
s1 = SentencePair(Sentence([u'B']), Sentence([u'Y']), aligned=True)
self.assertEquals(2, len(samples))
self.assertEquals([s0, s1], samples)
def test_empty_alignments(self):
A, B = [], []
samples = list(_aligned_samples(A, B, []))
self.assertEquals(0, len(samples))
def test_empty_alignments(self):
A, B = [], []
samples = list(_aligned_samples(A, B, []))
self.assertEquals(0, len(samples))
