示例#1
0
    def test_not_aligned_ranges(self):
        start = 5
        stop = 25
        step = 10
        range_set = RangeSet(start, stop, step)

        self.assertEqual(range_set.ranges, [Range(5, 9), Range(10, 19),
            Range(20, 25)])
示例#2
0
class RangeSetTestCase(unittest.TestCase):

    def setUp(self):
        self.start = 0
        self.stop = 49
        self.step = 10
        self.range_set = RangeSet(self.start, self.stop, self.step)

    def test_simple(self):
        workers = list(self.range_set.generate_workers())

        expected_workers = [MultiRangeWorker(0, 49, 10)]

        self.assertEqual(workers, expected_workers)

    def test_add_subrange(self):
        sub_range = SubRange(22, 25, value=42)
        self.range_set.add_sub_range(sub_range)

        workers = list(self.range_set.generate_workers())

        expected_range = Range(20, 29)
        expected_range.missing_ranges = [SubRange(20, 21),
            SubRange(26, 29)]
        expected_range.sub_ranges = [sub_range]
        partial_range_worker = RangeWorker(expected_range)

        expected_workers = [MultiRangeWorker(0, 19, 10),
            partial_range_worker, MultiRangeWorker(30, 49, 10)]
        self.assertEqual(workers, expected_workers)

    def test_not_aligned_ranges(self):
        start = 5
        stop = 25
        step = 10
        range_set = RangeSet(start, stop, step)

        self.assertEqual(range_set.ranges, [Range(5, 9), Range(10, 19),
            Range(20, 25)])
示例#3
0
 def setUp(self):
     self.start = 0
     self.stop = 49
     self.step = 10
     self.range_set = RangeSet(self.start, self.stop, self.step)