def test_dateToISO(self): """dateToISO should recurse properly""" d1 = {'k1':datetime.datetime(2012,12,21)} self.assertEqual({'k1': '2012-12-21T00:00:00'}, dm._dateToISO(d1)) d1 = {'k1':{'k2':datetime.datetime(2012,12,21)}} # regression, it does not traverse nested dicts self.assertEqual({'k1': {'k2': datetime.datetime(2012, 12, 21, 0, 0)}}, dm._dateToISO(d1)) d1 = {'k1':[datetime.datetime(2012,12,21), datetime.datetime(2012,12,22)] } self.assertEqual({'k1': ['2012-12-21T00:00:00', '2012-12-22T00:00:00']}, dm._dateToISO(d1)) d1 = datetime.datetime(2012,12,21) self.assertEqual('2012-12-21T00:00:00', dm._dateToISO(d1)) d1 = [datetime.datetime(2012,12,21), datetime.datetime(2012,12,22)] np.testing.assert_array_equal(dm._dateToISO(d1), ['2012-12-21T00:00:00', '2012-12-22T00:00:00'])
def test_dateToISOunaltered_dm(self): """Test to check that _dateToISO doesn't change datatypes, input dmarray""" data = spt.tickrange('20200101', '20200102', deltadays=datetime.timedelta(hours=1)).UTC exptype = type(data[0]) newdata = dm._dateToISO(data) restype = type(data[0]) self.assertEqual(exptype, restype)