from basetest import * class Thing: def __init__(self, timestamp): self.timestamp=timestamp def __str__(self): # age=now-self.timestamp struct=time.gmtime(self.timestamp) return("{}".format(time.strftime("%Y-%m-%d %H:%M:%S",struct))) class TestThinner(unittest2.TestCase): # def setUp(self): # return super().setUp() def test_incremental(self): ok=['2023-01-01 10:09:50', '2024-01-01 20:06:35', '2025-01-01 09:59:44', '2026-01-01 18:06:41', '2026-03-08 02:27:07', '2026-04-07 02:29:04', '2026-05-07 18:39:31', '2026-06-06 06:06:14', '2026-07-06 03:53:12', '2026-08-05 06:23:43', '2026-09-04 21:13:46', '2026-10-04 00:50:48', '2026-11-03 01:52:55', '2026-12-03 15:04:25', '2027-01-01 09:02:16', '2027-01-02 09:59:16', '2027-01-28 09:54:49', '2027-02-01 08:59:47', '2027-02-04 03:24:33', '2027-02-11 01:51:49', '2027-02-18 04:09:25', '2027-02-19 14:21:39', '2027-02-20 13:41:38', '2027-02-21 07:33:50', '2027-02-22 07:39:18', '2027-02-23 07:52:18', '2027-02-24 02:16:31', '2027-02-24 02:17:08', '2027-02-24 05:26:13', '2027-02-24 12:56:41'] #some arbitrary date now=1589229252 #we want deterministic results random.seed(1337) thinner=Thinner("5,10s1min,1d1w,1w1m,1m12m,1y5y") things=[] #thin incrementally while adding for i in range(0,5000): #increase random amount of time and maybe add a thing now=now+random.randint(0,3600*24) if random.random()>=0: things.append(Thing(now)) (keeps, removes)=thinner.thin(things, now=now) things=keeps result=[] for thing in things: result.append(str(thing)) print("Thinner result:") pprint.pprint(result) self.assertEqual(result, ok) def test_full(self): ok=['2022-02-24 15:54:37', '2023-01-01 10:09:50', '2024-01-01 20:06:35', '2025-01-01 09:59:44', '2026-01-01 18:06:41', '2026-03-01 23:23:58', '2026-03-08 02:27:07', '2026-04-07 02:29:04', '2026-05-07 18:39:31', '2026-06-06 06:06:14', '2026-07-06 03:53:12', '2026-08-05 06:23:43', '2026-09-04 21:13:46', '2026-10-04 00:50:48', '2026-11-03 01:52:55', '2026-12-03 15:04:25', '2027-01-01 09:02:16', '2027-01-02 09:59:16', '2027-01-25 20:00:35', '2027-01-28 09:54:49', '2027-02-01 08:59:47', '2027-02-04 03:24:33', '2027-02-11 01:51:49', '2027-02-18 04:09:25', '2027-02-19 14:21:39', '2027-02-20 13:41:38', '2027-02-21 07:33:50', '2027-02-22 07:39:18', '2027-02-23 07:52:18', '2027-02-24 02:16:31', '2027-02-24 02:17:08', '2027-02-24 05:26:13', '2027-02-24 12:56:41'] #some arbitrary date now=1589229252 #we want deterministic results random.seed(1337) thinner=Thinner("5,10s1min,1d1w,1w1m,1m12m,1y5y") things=[] for i in range(0,5000): #increase random amount of time and maybe add a thing now=now+random.randint(0,3600*24) if random.random()>=0: things.append(Thing(now)) (things, removes)=thinner.thin(things, now=now) result=[] for thing in things: result.append(str(thing)) print("Thinner result:") pprint.pprint(result) self.assertEqual(result, ok) if __name__ == '__main__': unittest.main()