def setUpClass(cls): # - GLOBAL CLASS FIXTURES - # put here any instruction you want to execute only *ONCE* *BEFORE* # executing *ALL* tests described below. _skip_if_no_project_id() test_requirements() clean_gbq_environment() gbq._Dataset(PROJECT_ID).create(DATASET_ID + "1")
def setUpClass(cls): # - GLOBAL CLASS FIXTURES - # put here any instruction you want to execute only *ONCE* *BEFORE* executing *ALL* tests # described below. if not PROJECT_ID: raise nose.SkipTest("Cannot run integration tests without a project id") test_requirements() clean_gbq_environment() gbq._Dataset(PROJECT_ID).create(DATASET_ID + "1")
def setUpClass(cls): # - GLOBAL CLASS FIXTURES - # put here any instruction you want to execute only *ONCE* *BEFORE* # executing *ALL* tests described below. _skip_if_no_project_id() _setup_common() clean_gbq_environment(_get_private_key_path()) gbq._Dataset(_get_project_id(), private_key=_get_private_key_path() ).create(DATASET_ID + "1")
def setUp(self): # - PER-TEST FIXTURES - # put here any instruction you want to be run *BEFORE* *EVERY* test is # executed. self.dataset = gbq._Dataset(PROJECT_ID) self.table = gbq._Table(PROJECT_ID, DATASET_ID + "1")
def test_list_table_zero_results(self): dataset_id = DATASET_ID + "2" self.dataset.create(dataset_id) table_list = gbq._Dataset(_get_project_id(), private_key=_get_private_key_path() ).tables(dataset_id) self.assertEqual(len(table_list), 0, 'Expected gbq.list_table() to return 0')
def setUp(self): # - PER-TEST FIXTURES - # put here any instruction you want to be run *BEFORE* *EVERY* test is # executed. self.dataset = gbq._Dataset(_get_project_id(), private_key=_get_private_key_path()) self.table = gbq._Table(_get_project_id(), DATASET_ID + "1", private_key=_get_private_key_path())
def clean_gbq_environment(private_key=None): dataset = gbq._Dataset(PROJECT_ID, private_key=private_key) for i in range(1, 10): if DATASET_ID + str(i) in dataset.datasets(): dataset_id = DATASET_ID + str(i) table = gbq._Table(PROJECT_ID, dataset_id, private_key=private_key) for j in range(1, 20): if TABLE_ID + str(j) in dataset.tables(dataset_id): table.delete(TABLE_ID + str(j)) dataset.delete(dataset_id)
def test_list_table_zero_results(self): dataset_id = DATASET_ID + "2" self.dataset.create(dataset_id) table_list = gbq._Dataset(PROJECT_ID).tables(dataset_id) self.assertEqual(len(table_list), 0, 'Expected gbq.list_table() to return 0')