def test_100(self): ft = FactorTable(conn, cik_dt='datetime', cik_iid='code', strict_cik=False) ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['size'], cik_dt='date', cik_iid='ip') # ft.set_cik_dt([datetime.datetime.now()]) res = ft.fetch(reduced=True, add_limit=True) self.assertEqual(res.shape[0], 100)
def test_add_factor_order1(self): ft = FactorTable(conn, strict_cik=False) ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['size'], cik_dt='date', cik_iid='ip') ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['idx'], cik_dt='date', cik_iid='ip') ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['norefer'], cik_dt='date', cik_iid='ip') s = ft.fetch(reduced=True, add_limit=True) self.assertTupleEqual(s.shape, (100, 3)) self.assertListEqual(s.columns.tolist(), ['size', 'idx', 'norefer', ])
def test_add_factor_from_ft(self): ft = FactorTable(conn, cik_dt='datetime', cik_iid='code', strict_cik=False) ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['size'], cik_dt='date', cik_iid='ip') ft2 = FactorTable(conn, cik_dt='datetime', cik_iid='code', strict_cik=False) ft2.add_factor('select * from EDGAR_LOG.edgar_log', factor_names=['norefer', 'size'], cik_dt='date', cik_iid='ip') ft.add_factor(ft2, factor_names=['norefer'], cik_dt='cik_dt', cik_iid='cik_iid') res = ft.fetch(reduced=True, add_limit=True) self.assertEqual(res.shape[0], 100)
# coding=utf-8 from ClickSQL.factor_table.factortable import FactorTable from test.conn import conn ft = FactorTable(conn, cik_dt='datetime', cik_iid='code', strict_cik=False) if __name__ == '__main__': ft.add_factor('EDGAR_LOG.edgar_log', factor_names=['size', 'ip', 'norefer'], ) query_func = ft._node print(query_func('desc EDGAR_LOG.edgar_log')) # Node.add_factor('EDGAR_LOG.edgar_log', factor_names='cik', cik_dt='date', cik_iid='ip') # Node.add_factor('test.test2', factor_names='test2', cik_dt='cik_dt', cik_iid='cik_iid') # f = ft.fetch(reduced=True, add_limit=True) # ft2 = FactorTable(conn, cik_dt='datetime', cik_iid='code', strict_cik=False) # ft.set_cik_dt([datetime.datetime.now()]) # res = ft.fetch(reduced=True, add_limit=True) # print(res) pass