def testSetReadConsole(self): yes = "yes\n" def sayyes(prompt): return yes rinterface.setReadConsole(sayyes) self.assertEquals(rinterface.getReadConsole(), sayyes) res = rinterface.baseNameSpaceEnv["readline"]() self.assertEquals(yes.strip(), res[0]) rinterface.setReadConsole(rinterface.consoleRead)
def testReadConsoleWithError(self): def f(prompt): raise Exception("Doesn't work.") rinterface.setReadConsole(f) tmp_file = tempfile.NamedTemporaryFile() stderr = sys.stderr sys.stderr = tmp_file try: res = rinterface.baseNameSpaceEnv["readline"]() except Exception, e: sys.stderr = stderr raise e
def tearDown(self): rinterface.setWriteConsole(rinterface.consolePrint) rinterface.setReadConsole(rinterface.consoleRead) rinterface.setReadConsole(rinterface.consoleFlush)