I would just make this a function and include in the test_gbq.py