def test_single_span(self): span = data.Span(self.file, 3, 5) fragments = renderer.fragmentize(self.file, [span]) self.assertEqual([f.content for f in fragments], [("012",), ("34",), ("56789",)]) self.assertEqual(fragments[0].spans, tuple()) self.assertEqual(fragments[1].spans, (span,)) self.assertEqual(fragments[2].spans, tuple())
def test_touching_spans(self): span_1 = data.Span(self.file, 3, 5) span_2 = data.Span(self.file, 5, 7) fragments = renderer.fragmentize(self.file, [span_1, span_2]) self.assertEqual([f.content for f in fragments], [("012",), ("34",), ("56",), ("789",)]) self.assertEqual(fragments[0].spans, tuple()) self.assertEqual(fragments[1].spans, (span_1,)) self.assertEqual(fragments[2].spans, (span_2,)) self.assertEqual(fragments[3].spans, tuple())
def test_completely_overlapping_spans(self): span_1 = data.Span(self.file, 1, 9) span_2 = data.Span(self.file, 4, 6) fragments = renderer.fragmentize(self.file, [span_1, span_2]) self.assertEqual([f.content for f in fragments], [("0",), ("123",), ("45",), ("678",), ("9",)]) self.assertEqual(fragments[0].spans, tuple()) self.assertEqual(fragments[1].spans, (span_1,)) self.assertEqual(set(fragments[2].spans), {span_1, span_2}) self.assertEqual(fragments[3].spans, (span_1,)) self.assertEqual(fragments[4].spans, tuple())
def test_overlapping_spans(self): span_1 = data.Span(self.file, 3, 7) span_2 = data.Span(self.file, 5, 9) fragments = renderer.fragmentize(self.file, [span_1, span_2]) self.assertEqual([f.content for f in fragments], [("012",), ("34",), ("56",), ("78",), ("9",)]) self.assertEqual(fragments[0].spans, tuple()) self.assertEqual(fragments[1].spans, (span_1,)) self.assertEqual(set(fragments[2].spans), {span_1, span_2}) self.assertEqual(fragments[3].spans, (span_2,)) self.assertEqual(fragments[4].spans, tuple())
def test_spans_are_ordered(self): # Check that the spans are ordered by span size span_1 = data.Span(self.file, 0, 3) span_2 = data.Span(self.file, 0, 2) span_3 = data.Span(self.file, 0, 1) fragments = renderer.fragmentize(self.file, [span_1, span_3, span_2]) self.assertEqual([f.content for f in fragments], [("0",), ("1",), ("2",), ("3456789",)]) self.assertEqual(fragments[0].spans, (span_1, span_2, span_3)) self.assertEqual(fragments[1].spans, (span_1, span_2)) self.assertEqual(fragments[2].spans, (span_1,)) self.assertEqual(fragments[3].spans, tuple()) span_1 = data.Span(self.file, 2, 3) span_2 = data.Span(self.file, 1, 3) span_3 = data.Span(self.file, 0, 3) fragments = renderer.fragmentize(self.file, [span_1, span_3, span_2]) self.assertEqual([f.content for f in fragments], [("0",), ("1",), ("2",), ("3456789",)]) self.assertEqual(fragments[0].spans, (span_3,)) self.assertEqual(fragments[1].spans, (span_3, span_2)) self.assertEqual(fragments[2].spans, (span_3, span_2, span_1)) self.assertEqual(fragments[3].spans, tuple())
def test_from_start_till_end(self): span = data.Span(self.file, 0, len(self.content)) fragments = renderer.fragmentize(self.file, [span]) self.assertEqual([f.content for f in fragments], [("0123456789",)]) self.assertEqual(fragments[0].spans, (span,))
def test_till_end(self): span = data.Span(self.file, 5, len(self.content)) fragments = renderer.fragmentize(self.file, [span]) self.assertEqual([f.content for f in fragments], [("01234",), ("56789",)]) self.assertEqual(fragments[0].spans, tuple()) self.assertEqual(fragments[1].spans, (span,))
def test_no_span(self): fragments = renderer.fragmentize(self.file, []) self.assertEqual([f.content for f in fragments], [("0123456789",)]) self.assertEqual(fragments[0].spans, tuple())