Python count_tornadoes Examples

Example #1

File: bigquery_tornadoes_test.py Project: gyamxxx/beam

 def test_basics(self):
     p = TestPipeline()
     rows = (p | 'create' >> beam.Create([{
         'month': 1,
         'day': 1,
         'tornado': False
     }, {
         'month': 1,
         'day': 2,
         'tornado': True
     }, {
         'month': 1,
         'day': 3,
         'tornado': True
     }, {
         'month': 2,
         'day': 1,
         'tornado': True
     }]))
     results = bigquery_tornadoes.count_tornadoes(rows)
     beam.assert_that(
         results,
         beam.equal_to([{
             'month': 1,
             'tornado_count': 2
         }, {
             'month': 2,
             'tornado_count': 1
         }]))
     p.run().wait_until_finish()

Example #2

File: bigquery_tornadoes_test.py Project: Sil1991/gcpdf-demo

 def test_basics(self):
     with TestPipeline() as p:
         rows = (p | 'create' >> beam.Create([{
             'month': 1,
             'day': 1,
             'tornado': False
         }, {
             'month': 1,
             'day': 2,
             'tornado': True
         }, {
             'month': 1,
             'day': 3,
             'tornado': True
         }, {
             'month': 2,
             'day': 1,
             'tornado': True
         }]))
         results = bigquery_tornadoes.count_tornadoes(rows)
         assert_that(
             results,
             equal_to([{
                 'month': 1,
                 'tornado_count': 2
             }, {
                 'month': 2,
                 'tornado_count': 1
             }]))

Example #3

File: bigquery_tornadoes_test.py Project: JavierRoger/beam

 def test_basics(self):
   with TestPipeline() as p:
     rows = (p | 'create' >> beam.Create([
         {'month': 1, 'day': 1, 'tornado': False},
         {'month': 1, 'day': 2, 'tornado': True},
         {'month': 1, 'day': 3, 'tornado': True},
         {'month': 2, 'day': 1, 'tornado': True}]))
     results = bigquery_tornadoes.count_tornadoes(rows)
     assert_that(results, equal_to([{'month': 1, 'tornado_count': 2},
                                    {'month': 2, 'tornado_count': 1}]))

Example #4

File: bigquery_tornadoes_test.py Project: dhalperi/incubator-beam

 def test_basics(self):
   p = TestPipeline()
   rows = (p | 'create' >> beam.Create([
       {'month': 1, 'day': 1, 'tornado': False},
       {'month': 1, 'day': 2, 'tornado': True},
       {'month': 1, 'day': 3, 'tornado': True},
       {'month': 2, 'day': 1, 'tornado': True}]))
   results = bigquery_tornadoes.count_tornadoes(rows)
   assert_that(results, equal_to([{'month': 1, 'tornado_count': 2},
                                  {'month': 2, 'tornado_count': 1}]))
   p.run().wait_until_finish()