def testSetFlushConsole(self): flush = {'count': 0} def f(): flush['count'] = flush['count'] + 1 rinterface.setFlushConsole(f) self.assertEquals(rinterface.getFlushConsole(), f) rinterface.baseNameSpaceEnv.get("flush.console")() self.assertEquals(1, flush['count']) rinterface.setWriteConsole(rinterface.consoleFlush)
def testFlushConsoleWithError(self): def f(prompt): raise Exception("Doesn't work.") rinterface.setFlushConsole(f) tmp_file = tempfile.NamedTemporaryFile() stderr = sys.stderr sys.stderr = tmp_file try: res = rinterface.baseNameSpaceEnv.get("flush.console")() except Exception, e: sys.stderr = stderr raise e