def test_urlbuilder_v4(self):
date_sequence = '2001 Mar 01'
ranger_output = _dateRanger(date_sequence)
gdeltstring_output = _gdeltRangeString(ranger_output, version=1)
urlbuilder_test = _urlBuilder(gdeltstring_output, version=1)
exp = 'http://data.gdeltproject.org/events/2001.zip'
return self.assertEqual(exp, urlbuilder_test, "Version 1 Url works.")
def test_urlbuilder_v2(self):
date_sequence = '2016 10 01'
ranger_output = _dateRanger(date_sequence)
gdeltstring_output = _gdeltRangeString(ranger_output, version=2)
urlbuilder_test = _urlBuilder(gdeltstring_output, version=2)
exp = 'http://data.gdeltproject.org/gdeltv2/20161001234500.export.CSV.zip'
return self.assertEqual(exp, urlbuilder_test, "Version 2 Url works.")
def test_urlbuilder_graph2_pass(self):
date_sequence = '2015 Apr 01'
ranger_output = _dateRanger(date_sequence)
gdeltstring_output = _gdeltRangeString(ranger_output, version=2)
urlbuilder_test = _urlBuilder(gdeltstring_output,
table='gkg',
version=2)
exp = 'http://data.gdeltproject.org/gdeltv2/20150401234500.gkg.csv.zip'
return self.assertEqual(exp, urlbuilder_test, "Version 2 Url works.")
def test_urlbuilder_v5(self):
date_sequence = '2013 Apr 01'
ranger_output = _dateRanger(date_sequence)
gdeltstring_output = _gdeltRangeString(ranger_output, version=1)
urlbuilder_test = _urlBuilder(gdeltstring_output,
table='gkg',
version=1)
exp = 'http://data.gdeltproject.org/gkg/20130401.gkg.csv.zip'
return self.assertEqual(exp, urlbuilder_test, "Version 1 Url works.")
def test_gdeltdate_urlbuilder_events_v1_fail_list(self):
date_sequence = ['201301', '201302']
exp = ("GDELT 1.0 Global Knowledge Graph requires dates greater than "
"or equal to April 1 2013")
with self.assertRaises(Exception) as context:
result = _urlBuilder(date_sequence, version=1, table='gkg')
the_exception = context.exception
return self.assertEqual(exp, str(the_exception),
"Version 1 old, gkg fail.")
def test_gdeltdate_urlbuilder_events_v1_three(self):
date_sequence = ['20160101234500', '20160102234500', '20160103234500']
exp = [
'http://data.gdeltproject.org/events/20160101234500.export.CSV.zip',
'http://data.gdeltproject.org/events/20160102234500.export.CSV.zip',
'http://data.gdeltproject.org/events/20160103234500.export.CSV.zip'
]
result = _urlBuilder(
['20160101234500', '20160102234500', '20160103234500'], version=1)
return self.assertEqual(exp, result, "Events 1 more than 2 failed.")
def test_urlbuilder_events1_passlist(self):
date_sequence = ['2015 Apr 01', '2015 Apr 02']
ranger_output = _dateRanger(date_sequence)
gdeltstring_output = _gdeltRangeString(ranger_output, version=1)
urlbuilder_test = _urlBuilder(gdeltstring_output,
table='events',
version=1)
exp = np.sort(
np.array([
'http://data.gdeltproject.org/events/20150401.export.CSV.zip',
'http://data.gdeltproject.org/events/20150402.export.CSV.zip'
]))
return np.testing.assert_array_equal(
np.sort(np.array(exp)), np.sort(np.array(urlbuilder_test)))
