def test_compute_top_sessions(self):
        p = TestPipeline()
        edits = p | beam.Create(self.EDITS)
        result = edits | top_wikipedia_sessions.ComputeTopSessions(1.0)

        beam.assert_that(result, beam.equal_to(self.EXPECTED))
        p.run()
  def test_compute_top_sessions(self):
    with TestPipeline() as p:
      edits = p | beam.Create(self.EDITS)
      result = edits | top_wikipedia_sessions.ComputeTopSessions(1.0)

      assert_that(result, equal_to(self.EXPECTED))