def test_gdeltrange_sequence_v1_2005(self):
date_sequence = ['2001 Feb 01', '2005 Feb 05']
ranger_output = _dateRanger(date_sequence)
gdeltstring_test = np.sort(
np.array(_gdeltRangeString(ranger_output, version=1)))
exp = np.sort(np.array(['2001', '2002', '2003', '2004', '2005']))
np.testing.assert_array_equal(exp, gdeltstring_test)
def test_gdeltrange_sequence_v2_single_tester(self):
date_sequence = ['2016 10 01', '2016 Dec 31']
ranger_output = _dateRanger(date_sequence)
gdeltstring_test = _gdeltRangeString(ranger_output,
coverage=True,
version=2)
return self.assertIsInstance(gdeltstring_test, np.ndarray)
def test_gdeltrange_nearest_single(self):
date_string = _dateRanger([str((datetime.datetime.now()).date())])
ranger_test = _gdeltRangeString(date_string, coverage=False, version=2)
# build time range including 30 minutes before and 30 after current
r = pd.date_range(
datetime.datetime.now() - datetime.timedelta(minutes=30),
datetime.datetime.now() + datetime.timedelta(minutes=30),
freq='15min')
return self.assertTrue(r[0] < parse(ranger_test) < r[-1])
def test_gdeltrange_nearest_full(self):
date_string = _dateRanger([str((datetime.datetime.now()).date())])
ranger_test = _gdeltRangeString(date_string, coverage=True, version=2)
# convert result to array of python datetimes
vect = np.vectorize(parse)
comp = vect(ranger_test)
# build time range of all day up to latest 15 minute interval
# convert to pydatetime array
exp = (pd.date_range(start=str(datetime.datetime.now().date()),
periods=len(ranger_test), freq='15 min')).\
to_pydatetime()
# compare arrays
return np.testing.assert_array_equal(comp, exp, 'Not equal')
def test_gdeltrange_sequence_v2_single_string(self):
date_sequence = '2016 10 01'
ranger_output = _dateRanger(date_sequence)
gdeltstring_test = _gdeltRangeString(ranger_output, version=2)
exp = '20161001234500'
return self.assertEqual(exp, gdeltstring_test)
