def test_init(self): td_fdw = self.td_fdw self.assertEqual(td_fdw.apikey, 'apikey1234') self.assertEqual(td_fdw.database, 'foodb') self.assertEqual(td_fdw.table, 'bartbl') self.assertEqual(td_fdw.query_engine, 'presto') self.assertEqual(td_fdw.columns, ['name', 'age']) td_fdw = TreasureDataFdw( { 'apikey': 'apikey1234', 'database': 'foodb', 'query': 'SELECT code, COUNT(1) from access_log GROUP BY code', 'query_engine': 'hive', }, ['name', 'age']) self.assertEqual(td_fdw.apikey, 'apikey1234') self.assertEqual(td_fdw.database, 'foodb') self.assertEqual( td_fdw.query, 'SELECT code, COUNT(1) from access_log GROUP BY code') self.assertEqual(td_fdw.query_engine, 'hive') self.assertEqual(td_fdw.columns, ['name', 'age']) td_fdw = TreasureDataFdw({}, []) self.assertEqual(td_fdw.apikey, None) self.assertEqual(td_fdw.database, None) self.assertEqual(td_fdw.query, None) self.assertEqual(td_fdw.query_engine, 'presto') self.assertEqual(td_fdw.columns, [])
def setUp(self): self.td_fdw = TreasureDataFdw( { 'apikey': 'apikey1234', 'database': 'foodb', 'table': 'bartbl' }, ['name', 'age'])