コード例 #1
0
ファイル: iobase_test.py プロジェクト: piter75/apache-beam
 def test_try_split_with_any_exception(self):
     source_bundle = SourceBundle(
         range_trackers.OffsetRangeTracker.OFFSET_INFINITY,
         RangeSource(0, range_trackers.OffsetRangeTracker.OFFSET_INFINITY),
         0, range_trackers.OffsetRangeTracker.OFFSET_INFINITY)
     self.sdf_restriction_tracker = (
         iobase._SDFBoundedSourceRestrictionTracker(
             iobase._SDFBoundedSourceRestriction(source_bundle)))
     self.sdf_restriction_tracker.try_claim(0)
     self.assertIsNone(self.sdf_restriction_tracker.try_split(0.5))
コード例 #2
0
ファイル: iobase_test.py プロジェクト: piter75/apache-beam
 def setUp(self):
     self.initial_start_pos = 0
     self.initial_stop_pos = 4
     source_bundle = SourceBundle(
         self.initial_stop_pos - self.initial_start_pos,
         RangeSource(self.initial_start_pos, self.initial_stop_pos),
         self.initial_start_pos, self.initial_stop_pos)
     self.sdf_restriction_tracker = (
         iobase._SDFBoundedSourceRestrictionTracker(
             iobase._SDFBoundedSourceRestriction(source_bundle)))
コード例 #3
0
ファイル: iobase_test.py プロジェクト: piter75/apache-beam
 def test_create_tracker(self):
     expected_start = 1
     expected_stop = 3
     source_bundle = SourceBundle(expected_stop - expected_start,
                                  RangeSource(1, 3), expected_start,
                                  expected_stop)
     restriction_tracker = (self.sdf_restriction_provider.create_tracker(
         iobase._SDFBoundedSourceRestriction(source_bundle)))
     self.assertTrue(
         isinstance(restriction_tracker,
                    iobase._SDFBoundedSourceRestrictionTracker))
     self.assertEqual(expected_start, restriction_tracker.start_pos())
     self.assertEqual(expected_stop, restriction_tracker.stop_pos())