def test_sleep2(self):
s = HistoricalScheduler()
n = [0]
def action(scheduler, state):
s.sleep(timedelta(3 * 6000))
n[0] += 1
s.schedule_absolute(s.now + timedelta(6000), action)
s.schedule_absolute(s.now + timedelta(6000), action)
s.advance_to(s.now + timedelta(5 * 6000))
self.assertEqual(2, n[0])
def test_sleep2(self):
s = HistoricalScheduler()
n = [0]
def action(scheduler, state):
s.sleep(timedelta(3 * 6000))
n[0] += 1
s.schedule_absolute(s.now + timedelta(6000), action)
s.schedule_absolute(s.now + timedelta(6000), action)
s.advance_to(s.now + timedelta(5 * 6000))
self.assertEqual(2, n[0])
def test_with_comparer(self):
now = datetime.utcnow()
def reverse_comparer (x, y):
return y < x
s = HistoricalScheduler(now, reverse_comparer)
res = []
s.schedule_absolute(now - timedelta(1000), lambda a, b: res.append(1))
s.schedule_absolute(now - timedelta(2000), lambda a, b: res.append(2))
s.start()
assert_equals(res, [1,2])
def test_cancellation(self):
s = HistoricalScheduler()
list = []
d = s.schedule_absolute(
time(2), lambda a, b: list.append(Timestamped(2, s.now)))
def action(scheduler, state):
list.append(Timestamped(0, s.now))
d.dispose()
s.schedule_absolute(time(1), action)
s.start()
assert_equals(list, [Timestamped(0, time(1))])
def test_cancellation(self):
s = HistoricalScheduler()
list = []
d = s.schedule_absolute(time(2), lambda a,b: list.append(Timestamped(2, s.now)))
def action(scheduler, state):
list.append(Timestamped(0, s.now))
d.dispose()
s.schedule_absolute(time(1), action)
s.start()
assert_equals(list, [
Timestamped(0, time(1))
])
def test_order(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(2),
lambda a, b: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(3),
lambda a, b: list.append(Timestamped(3, s.now)))
s.schedule_absolute(time(1),
lambda a, b: list.append(Timestamped(0, s.now)))
s.schedule_absolute(time(1),
lambda a, b: list.append(Timestamped(1, s.now)))
s.start()
assert_equals(list, [
Timestamped(0, time(1)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(3, time(3))
])
def test_order(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(2), lambda a, b: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(3), lambda a, b: list.append(Timestamped(3, s.now)))
s.schedule_absolute(time(1), lambda a, b: list.append(Timestamped(0, s.now)))
s.schedule_absolute(time(1), lambda a, b: list.append(Timestamped(1, s.now)))
s.start()
assert_equals(list, [
Timestamped(0, time(1)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(3, time(3))
])
def test_start_stop(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(0), lambda sc,st: list.append(Timestamped(1, s.now)))
s.schedule_absolute(time(1), lambda sc,st: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(2), lambda sc,st: s.stop())
s.schedule_absolute(time(3), lambda sc,st: list.append(Timestamped(3, s.now)))
s.schedule_absolute(time(4), lambda sc,st: s.stop())
s.schedule_absolute(time(5), lambda sc,st: s.start())
s.schedule_absolute(time(6), lambda sc,st: list.append(Timestamped(4, s.now)))
s.start()
self.assertEqual(time(2), s.now)
self.assertEqual(time(2), s.clock)
s.start()
self.assertEqual(time(4), s.now)
self.assertEqual(time(4), s.clock)
s.start()
self.assertEqual(time(6), s.now)
self.assertEqual(time(6), s.clock)
s.start()
self.assertEqual(time(6), s.now)
self.assertEqual(time(6), s.clock)
assert_equals(list, [
Timestamped(1, time(0)),
Timestamped(2, time(1)),
Timestamped(3, time(3)),
Timestamped(4, time(6))
])
def test_advance_by(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(0), lambda a, b: list.append(Timestamped(0, s.now)))
s.schedule_absolute(time(1), lambda a, b: list.append(Timestamped(1, s.now)))
s.schedule_absolute(time(2), lambda a, b: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(10), lambda a, b: list.append(Timestamped(10, s.now)))
s.schedule_absolute(time(11), lambda a, b: list.append(Timestamped(11, s.now)))
s.advance_by(time(8) - s.now)
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.schedule_absolute(time(7), lambda a, b: list.append(Timestamped(7, s.now)))
s.schedule_absolute(time(8), lambda a, b: list.append(Timestamped(8, s.now)))
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.advance_by(timedelta(0))
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.advance_by(from_days(2))
self.assertEqual(time(10), s.now)
self.assertEqual(time(10), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(7, time(8)),
Timestamped(8, time(8)),
Timestamped(10, time(10))
])
s.advance_by(from_days(90))
self.assertEqual(time(100), s.now)
self.assertEqual(time(100), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(7, time(8)),
Timestamped(8, time(8)),
Timestamped(10, time(10)),
Timestamped(11, time(11))
])
def test_start_stop(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(0), lambda sc,st: list.append(Timestamped(1, s.now)))
s.schedule_absolute(time(1), lambda sc,st: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(2), lambda sc,st: s.stop())
s.schedule_absolute(time(3), lambda sc,st: list.append(Timestamped(3, s.now)))
s.schedule_absolute(time(4), lambda sc,st: s.stop())
s.schedule_absolute(time(5), lambda sc,st: s.start())
s.schedule_absolute(time(6), lambda sc,st: list.append(Timestamped(4, s.now)))
s.start()
self.assertEqual(time(2), s.now)
self.assertEqual(time(2), s.clock)
s.start()
self.assertEqual(time(4), s.now)
self.assertEqual(time(4), s.clock)
s.start()
self.assertEqual(time(6), s.now)
self.assertEqual(time(6), s.clock)
s.start()
self.assertEqual(time(6), s.now)
self.assertEqual(time(6), s.clock)
assert_equals(list, [
Timestamped(1, time(0)),
Timestamped(2, time(1)),
Timestamped(3, time(3)),
Timestamped(4, time(6))
])
def test_advance_by(self):
s = HistoricalScheduler()
list = []
s.schedule_absolute(time(0), lambda a, b: list.append(Timestamped(0, s.now)))
s.schedule_absolute(time(1), lambda a, b: list.append(Timestamped(1, s.now)))
s.schedule_absolute(time(2), lambda a, b: list.append(Timestamped(2, s.now)))
s.schedule_absolute(time(10), lambda a, b: list.append(Timestamped(10, s.now)))
s.schedule_absolute(time(11), lambda a, b: list.append(Timestamped(11, s.now)))
s.advance_by(time(8) - s.now)
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.schedule_absolute(time(7), lambda a, b: list.append(Timestamped(7, s.now)))
s.schedule_absolute(time(8), lambda a, b: list.append(Timestamped(8, s.now)))
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.advance_by(timedelta(0))
self.assertEqual(time(8), s.now)
self.assertEqual(time(8), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2))
])
s.advance_by(from_days(2))
self.assertEqual(time(10), s.now)
self.assertEqual(time(10), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(7, time(8)),
Timestamped(8, time(8)),
Timestamped(10, time(10))
])
s.advance_by(from_days(90))
self.assertEqual(time(100), s.now)
self.assertEqual(time(100), s.clock)
assert_equals(list, [
Timestamped(0, time(0)),
Timestamped(1, time(1)),
Timestamped(2, time(2)),
Timestamped(7, time(8)),
Timestamped(8, time(8)),
Timestamped(10, time(10)),
Timestamped(11, time(11))
])
