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_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_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 span(self, start, end):
file = data.Submission(".", ["bar/foo"]).files[0]
return data.Span(file, start, end)
def span(self, start, end):
return data.Span(self.file, start, end)