self.client.GetJobReference, invalid) def testRaiseError(self): # Confirm we handle arbitrary errors gracefully. try: bigquery_client.BigqueryClient.RaiseError({}) except bigquery_client.BigqueryError as _: pass def testJsonToInsertEntry(self): result = [ bigquery_client.JsonToInsertEntry(None, '{"a":1}'), bigquery_client.JsonToInsertEntry('key', '{"b":2}'), ] self.assertEquals([None, 'key'], [x[0] for x in result]) self.assertEquals(1, result[0][1]['a']) self.assertEquals(2, result[1][1]['b']) self.assertRaisesRegexp(bigquery_client.BigqueryClientError, r'Could not parse', bigquery_client.JsonToInsertEntry, None, '_junk_') self.assertRaisesRegexp(bigquery_client.BigqueryClientError, r'not a JSON object', bigquery_client.JsonToInsertEntry, None, '[1, 2]') if __name__ == '__main__': googletest.main()
class CsvFormatterTest(TableFormatterTest): def setUp(self): self.format_class = table_formatter.CsvFormatter super(CsvFormatterTest, self).setUp() def testStr(self): table_repr = '\n'.join(( 'foo,longer header', 'a,3', 'abc,123')) self.assertEquals(table_repr, str(self.formatter)) class NullFormatterTest(TableFormatterTest): def setUp(self): self.format_class = table_formatter.NullFormatter super(NullFormatterTest, self).setUp() def testStr(self): self.assertEquals('', str(self.formatter)) def testUnicodeRow(self): self.assertEquals('', unicode(self.formatter)) if __name__ == '__main__': googletest.main()