def run_stats(geo_years, id_types, base_query): response_dict = {} for geo_year in geo_years: print "\n\n****\n", geo_year for id_type in id_types: response_dict[(id_type, geo_year)] = get_from_query_gds_in_pmc_fulltext_dict(id_type, base_query, geo_year) pkl_file = open("scienceplot_new/results/" + id_type + "_dict" + geo_year + "b.pkl", "wb") pickle.dump(response_dict[(id_type, geo_year)], pkl_file) pkl_file.close() fh = open("scienceplot_new/results/" + id_type + geo_year + "b.csv", "w") header = ",".join(fields) fh.write(header + "\n") dataset.csv_write_to_file(fh, response_dict[(id_type, geo_year)].values()) fh.close() return(response_dict)
def test_csv_file(self): import StringIO string_buffer = StringIO.StringIO() input = test_data_csv_format response = dataset.csv_write_to_file(string_buffer, input) response = string_buffer.getvalue() string_buffer.close() assert_equals(response, test_data_csv_string)
def run_stats(geo_years, id_types, base_query): response_dict = {} for geo_year in geo_years: print "\n\n****\n", geo_year for id_type in id_types: response_dict[( id_type, geo_year)] = get_from_query_gds_in_pmc_fulltext_dict( id_type, base_query, geo_year) pkl_file = open( "scienceplot_new/results/" + id_type + "_dict" + geo_year + "b.pkl", "wb") pickle.dump(response_dict[(id_type, geo_year)], pkl_file) pkl_file.close() fh = open( "scienceplot_new/results/" + id_type + geo_year + "b.csv", "w") header = ",".join(fields) fh.write(header + "\n") dataset.csv_write_to_file( fh, response_dict[(id_type, geo_year)].values()) fh.close() return (response_dict)