def test_scheduler_early_exhaustion(self):
procs = 2
streams = [["%s_a" % n, "%s_b" % n] for n in range(procs)]
for events in intermerge(*streams):
result = timelimited(2, self.scheduler_basic, procs, events[:len(events)/2])
self.assertEqual(events[:len(events)/2], result[:len(events)/2])
self.assertEqual(sorted(events), sorted(result))
def wrapped(*args, **kwargs):
modules = already_imported(*self.modules)
if modules:
raise IsolationException(
"Module %s was already imported" % ', '.join(modules))
if self.spawn_process:
return timelimited(None, func, *args, **kwargs)
else:
return func(*args, **kwargs)
def __call__(self, *args, **kwargs):
return timelimited(None, self._func, *args, **kwargs)
def test_ending(self):
self.assertEqual(timelimited(1, ends, 2), 2**2)
def test_scheduler_example(self):
procs = 2
streams = [["%s_a" % n, "%s_b" % n] for n in range(procs)]
for events in intermerge(*streams):
result = timelimited(2, self.scheduler_basic, procs, events)
self.assertEqual(events, result)
