def test_row(self):
from google.cloud.bigquery.table import Row
VALUES = (1, 2, 3)
row = Row(VALUES, {'a': 0, 'b': 1, 'c': 2})
self.assertEqual(row.a, 1)
self.assertEqual(row[1], 2)
self.assertEqual(row['c'], 3)
self.assertEqual(len(row), 3)
self.assertEqual(row.values(), VALUES)
self.assertEqual(set(row.keys()), set({'a': 1, 'b': 2, 'c': 3}.keys()))
self.assertEqual(set(row.items()),
set({'a': 1, 'b': 2, 'c': 3}.items()))
self.assertEqual(row.get('a'), 1)
self.assertEqual(row.get('d'), None)
self.assertEqual(row.get('d', ''), '')
self.assertEqual(row.get('d', default=''), '')
self.assertEqual(repr(row),
"Row((1, 2, 3), {'a': 0, 'b': 1, 'c': 2})")
self.assertFalse(row != row)
self.assertFalse(row == 3)
with self.assertRaises(AttributeError):
row.z
with self.assertRaises(KeyError):
row['z']
def create_bq_payload(data: Row):
"""
Takes in a row object from BigQuery and returns a dictionary
of key value pairs based on the row columns and values
Args:
data: The Row object from the BigQuery API
Returns:
A python dictionary object
"""
results = {k:v for k, v in data.items()}
return results