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