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)