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)