def test_small_task(self):
        executor = ThreadPoolExecutor(1)

        def task(arg):
            return 'OK %s' % arg

        f = executor.submit(task, 'ARG')
        assert f.result(0.01) == 'OK ARG'
    def test_task_failure(self):
        class MyException(Exception):
            pass

        executor = ThreadPoolExecutor(1)

        def task(arg):
            raise MyException

        f = executor.submit(task, 'ARG')
        assert isinstance(f.exception(0.01), MyException)