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))