beam-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "Robert Bradshaw (JIRA)" <j...@apache.org>
Subject [jira] [Created] (BEAM-1450) NewDoFn does not properly explode windows
Date Thu, 09 Feb 2017 20:16:41 GMT
Robert Bradshaw created BEAM-1450:
-------------------------------------

             Summary: NewDoFn does not properly explode windows
                 Key: BEAM-1450
                 URL: https://issues.apache.org/jira/browse/BEAM-1450
             Project: Beam
          Issue Type: Bug
          Components: sdk-py
            Reporter: Robert Bradshaw
            Assignee: Robert Bradshaw


A window-taking DoFn's process gets called once per window, but each output gets placed into
all windows. 

E.g. the following test fails

  def test_window_param(self):
    class TestDoFn(DoFn):
      def process(self, element, window=DoFn.WindowParam):
        yield (element, (float(window.start), float(window.end)))

    pipeline = TestPipeline()
    pcoll = (pipeline
             | Create([1, 7])
             | Map(lambda x: TimestampedValue(x, x))
             | WindowInto(windowfn=SlidingWindows(10, 5))
             | ParDo(TestDoFn()))
    assert_that(pcoll, equal_to([(1, (-5, 5)), (1, (0, 10)),
                                 (7, (0, 10)), (7, (5, 15))]))
    pcoll2 = pcoll | 'Again' >> ParDo(TestDoFn())
    assert_that(
        pcoll2,
        equal_to([
            ((1, (-5, 5)), (-5, 5)), ((1, (0, 10)), (0, 10)),
            ((7, (0, 10)), (0, 10)), ((7, (5, 15)), (5, 15))]),
        label='doubled windows')
    pipeline.run()




--
This message was sent by Atlassian JIRA
(v6.3.15#6346)

Mime
View raw message