def test_with_spaces(self):
self.assertEqual(
list(PiecewiseRange('0-0, 4-8, 20-21, 43-45')),
[0, 4, 5, 6, 7, 8, 20, 21, 43, 44, 45],
)
def test_looping_multiple_times(self):
numbers = PiecewiseRange('1-3,5-8')
self.assertEqual(list(numbers), [1, 2, 3, 5, 6, 7, 8])
self.assertEqual(list(numbers), [1, 2, 3, 5, 6, 7, 8])
def test_single_digit_range_and_ranges(self):
self.assertEqual(
list(PiecewiseRange('1-2,4,8-10')),
[1, 2, 4, 8, 9, 10],
)
def test_three_ranges(self):
self.assertEqual(
list(PiecewiseRange('1-3,4-6,8-10')),
[1, 2, 3, 4, 5, 6, 8, 9, 10],
)
def test_equality(self):
self.assertEqual(repr(PiecewiseRange('1-3')), "PiecewiseRange('1-3')")
self.assertEqual(
PiecewiseRange('1-2, 6-8, 12-14'),
PiecewiseRange('1-2, 6-8, 12-14'),
)
self.assertEqual(
PiecewiseRange('1-2,\t\r6-8,\n 12-14'),
PiecewiseRange('1-2, \t6-8, 12-14\t'),
)
self.assertEqual(
repr(PiecewiseRange('1,2,3')),
repr(PiecewiseRange('1-3')),
)
self.assertEqual(
PiecewiseRange('1-2,6,7,8-10, 12, 13, 14'),
PiecewiseRange('1-2, 6-10, 12-14'),
)
self.assertNotEqual(
PiecewiseRange('1-2,6,7,8-10, 12, 13, 15'),
PiecewiseRange('1-2, 6-10, 12-14'),
)
self.assertFalse(PiecewiseRange('1-3') == PiecewiseRange('1-2,4'))
self.assertFalse(PiecewiseRange('1-3') != PiecewiseRange('1-2,3'))
self.assertEqual(
PiecewiseRange('1-999999999,1000000000'),
PiecewiseRange('1-1000000000'),
)
def test_indexable(self):
self.assertEqual(PiecewiseRange('1-2, 6-8, 12-14')[5], 12)
self.assertEqual(PiecewiseRange('1-2, 6, 12-14')[2], 6)
self.assertEqual(PiecewiseRange('1-2, 6-8, 12-14')[-2], 13)
def test_len(self):
self.assertEqual(len(PiecewiseRange('1-4, 8, 11-14')), 9)
