def cleanup_test_table():
yield
client = create_client()
try:
client.delete_table(f'{DATASET}.test', not_found_ok=True)
except NotFound:
pass
from bq import create_client, read_sql, query
DATASET = 'ai2_replication'
client = create_client()
def make_table(table, **kw):
sql = read_sql(f'../ai2_replication/{table}.sql')
job = query(sql, table, dataset=DATASET, truncate=True, **kw)
return job.result()
make_table('institutions')
make_table('paper_authors_w_countries')
make_table('language')
make_table('ai_papers_any_author')
make_table('paper_author_institution')
make_table('oecd_comparison')
def test_create_client():
# We shouldn't get a UserWarning about using GCP user credentials
with pytest.warns(None) as warnings:
client = create_client()
assert len(warnings) == 0
assert client.project == PROJECT_ID
def client():
return create_client()