def create_agg_feat_table(method, table, num_cols):
execute_sql(method)
df_test = pd.read_sql("""SELECT * FROM """ + table + """ WHERE
distinct_id = '681CFA45-B526-4FE4-A2DF-F73F2D0CF674'""", engine)
n.assert_equal(len(df_test), 1)
n.assert_equal(len(df_test.columns), num_cols)
check_one_row_per_user(table)
def test_create_users_all_story_features():
first = 0
last = 0
chunk_size = 30
for i in range(NUM_STORIES / chunk_size + 1):
first = i * chunk_size
last = (i + 1) * chunk_size
if last > NUM_STORIES:
last = NUM_STORIES
sql = c.create_users_all_story_features(i, first, last)
sql = tests.replace_dbs(sql)
engine.execute(sql)
table = "users_all_story_features_" + str(i)
df_test = pd.read_sql("""SELECT * FROM """ + table + """ WHERE
distinct_id = '681CFA45-B526-4FE4-A2DF-F73F2D0CF674'""", engine)
n.assert_equal(len(df_test), 1)
n.assert_equal(len(df_test.columns), (last - first) * 1 + 1)
check_one_row_per_user(table)
