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 ""