Elgato_dark/zeppelin
1
0
Fork 0
mirror of https://github.com/apache/zeppelin synced 2026-05-24 09:38:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

make z.getInterpreterContext().out().clear() work in PythonInterpreter

Browse source
This commit is contained in:
Lee moon soo 2018-03-02 16:50:22 -08:00
parent 21dc20d886
commit 224a2df8e6
3 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/src/main/java/org/apache/zeppelin/python/PythonInterpreter.java
View file

@ -388,6 +388,7 @@ public class PythonInterpreter extends Interpreter implements ExecuteResultHandl
zeppelinContext.setGui(context.getGui());
zeppelinContext.setNoteGui(context.getNoteGui());
zeppelinContext.setInterpreterContext(context);
if (!pythonscriptRunning) {
return new InterpreterResult(Code.ERROR, "python process not running"

2
python/src/main/resources/python/zeppelin_python.py
View file

@ -61,7 +61,7 @@ class PyZeppelinContext(object):
self._setup_matplotlib()
def getInterpreterContext(self):
return self.z.getCurrentInterpreterContext()
return self.z.getInterpreterContext()
def input(self, name, defaultValue=""):
return self.z.input(name, defaultValue)

6
python/src/test/java/org/apache/zeppelin/python/PythonInterpreterTest.java
View file

@ -123,6 +123,12 @@ public class PythonInterpreterTest implements InterpreterOutputListener {
assertEquals(InterpreterResult.Code.SUCCESS, pythonInterpreter.interpret(pyValidCode, context).code());
}
@Test
public void testOutputClear() throws InterpreterException {
InterpreterResult result = pythonInterpreter.interpret("print(\"Hello\")\nz.getInterpreterContext().out().clear()\nprint(\"world\")\n", context);
assertEquals("%text world\n", out.getCurrentOutput().toString());
}
@Override
public void onUpdateAll(InterpreterOutput out) {