def test_range_set_to_list(self):
expected = [
3,
]
result = range_set_to_list('3')
self.assertEqual(expected, result)
expected = [3, 4, 5]
result = range_set_to_list('3-5')
self.assertEqual(expected, result)
expected = [3, 4, 5, 6, 7, 8, 9]
result = range_set_to_list('3-5,9,4,5-8')
self.assertEqual(expected, result)
def __init__(self, code_box, spec, highlight_on):
self.code_box = code_box
self.highlight_on = highlight_on
if isinstance(spec, str):
self.numbers = range_set_to_list(spec)
else:
self.numbers = spec
def steps(self):
all_steps = []
previous_numbers = None
for spec in self.spec_list:
if isinstance(spec, str):
numbers = range_set_to_list(spec)
else:
numbers = spec
if previous_numbers:
all_steps.append( steplib.HighlightLines(self.code_box,
previous_numbers, False) )
all_steps.extend([
steplib.HighlightLines(self.code_box, numbers, True),
steplib.CellEnd()
])
previous_numbers = numbers
return all_steps