def test_four(): """Return eventid and summary for all events with over 500 casualties.""" datalog_query = """q(eventid,summary) :- gtd(eventid, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, summary, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, nkill, _, _, nwound, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), nkill+nwound > 500""" wrapper = mysql_wrapper.MysqlWrapper() datalog_df = wrapper.execute(datalog_query) mysql_query = """select g.eventid, g.summary from gtd.gtd g where g.nkill + g.nwound > 500 """ mysql_df = execute_test_sql(mysql_query) assert_frame_equal(datalog_df, mysql_df) # mysql_df.equals(datalog_df)
def test_two(): """Return the summary of all events in the city of Khost, Afghanistan in 2012.""" datalog_query = """q(summary) :- gtd(_, iyear, _, _, _, _, _, _, countrytxt, _, _, _, city, _, _, _, _, _, summary, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), countrytxt = 'Afghanistan', city = 'Khost', iyear = 2012""" wrapper = mysql_wrapper.MysqlWrapper() datalog_df = wrapper.execute(datalog_query) mysql_query = """select summary from gtd.gtd g where country_txt = 'Afghanistan' and city = 'Khost' and iyear = 2012""" mysql_df = execute_test_sql(mysql_query) assert_frame_equal(datalog_df, mysql_df) # mysql_df.equals(datalog_df)
def test_one(): """Return the eventid's that occurred on 2012/12/25.""" datalog_query = """q(eventid) :- gtd(eventid, iyear, imonth, iday, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,_, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _),iyear = 2012, imonth = 12, iday = 25""" wrapper = mysql_wrapper.MysqlWrapper() datalog_df = wrapper.execute(datalog_query) mysql_query = """select eventid from gtd.gtd g where iyear = 2012 and imonth = 12 and iday = 25""" mysql_df = execute_test_sql(mysql_query) assert_frame_equal(datalog_df, mysql_df) # mysql_df.equals(datalog_df)
def test_three(): """Which countries had events on 2012-09-11?""" datalog_query = """q(country_txt) :- gtd(_, iyear, imonth, iday, _, _, _, _, country_txt, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), iday = 11, imonth = 9, iyear = 2012, DISTINCT """ wrapper = mysql_wrapper.MysqlWrapper() datalog_df = wrapper.execute(datalog_query) mysql_query = """select distinct country_txt from gtd.gtd g where g.imonth = 9 and g.iyear = 2012 and g.iday = 11 """ mysql_df = execute_test_sql(mysql_query) assert_frame_equal(datalog_df, mysql_df) # mysql_df.equals(datalog_df)
def test_five(): """Which 5 days (month and day only) have the most attacks over all the years?""" datalog_query = """q(iday, imonth,num_events) :- gtd(eventid, _, imonth, iday, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), GROUP_BY([iday,imonth], num_events = COUNT(eventid)), SORT_BY(num_events, 'DESC'), LIMIT(5)""" wrapper = mysql_wrapper.MysqlWrapper() datalog_df = wrapper.execute(datalog_query) mysql_query = """select g.iday, g.imonth,count(eventid) num_events from gtd.gtd g group by g.iday, g.imonth order by count(eventid) desc limit 5; """ mysql_df = execute_test_sql(mysql_query) assert_frame_equal(datalog_df, mysql_df) # mysql_df.equals(datalog_df)
T1<= 2014, T1 >= 2013" ## Query 5 q5 = "meta_DB1(I1a,Y1,M1,D1,C1,g1,gs1) :- gtd(I1a, 2016, M1, D1, _, _, _, _, C1, _, _, _, _, _, _, \ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, g1, gs1, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _)" ## Query with all functions q6 = "DB1(s1,p) :- gtd(_, T1, 12, d, _, _, _, _,'Iraq', _, _, _, _, _, _, _, _, _, s1, _, \ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, \ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), T1<= 2014, GROUP_BY([s1], p = MAX(d)), SORT_BY (p, 'DESC'), p > 40,LIMIT (10)" # ## Query with all functions q7 = "DB1(s1,p) :- gtd(_, T1, 12, d, s2, _, _, _,'Iraq', _, _, _, _, _, _, _, _, _, s1, _, \ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _,\ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, \ _, _, _, _, _, _, _, _, _, _, _, _, _, _, _), s2 > 10, T1<= 2014, GROUP_BY([s1], p = MAX(d)),SORT_BY (p, 'DESC'), LIMIT (10)" queries = [q1, q2, q3, q4, q5, q6, q7] for query in queries: wrapper = mysql_wrapper.MysqlWrapper() print query print "" dataframe = wrapper.execute(query) print dataframe print ""