Examples with InterpreterResult org.apache.zeppelin.interpreter.InterpreterResult used on opensource projects

Example 26 with InterpreterResult

use of org.apache.zeppelin.interpreter.InterpreterResult in project zeppelin by apache.

the class PythonInterpreterPandasSqlTest method errorMessageIfDependenciesNotInstalled.

@Test
public void errorMessageIfDependenciesNotInstalled() {
    InterpreterResult ret;
    ret = sql.interpret("SELECT * from something", context);
    assertNotNull(ret);
    assertEquals(ret.message().get(0).getData(), InterpreterResult.Code.ERROR, ret.code());
    assertTrue(ret.message().get(0).getData().contains("no such table: something"));
}

Example 27 with InterpreterResult

use of org.apache.zeppelin.interpreter.InterpreterResult in project zeppelin by apache.

the class PythonInterpreterPandasSqlTest method sqlOverTestDataPrintsTable.

@Test
public void sqlOverTestDataPrintsTable() throws IOException {
    InterpreterResult ret;
    // given
    //String expectedTable = "name\tage\n\nmoon\t33\n\npark\t34";
    ret = python.interpret("import pandas as pd", context);
    ret = python.interpret("import numpy as np", context);
    // DataFrame df2 \w test data
    ret = python.interpret("df2 = pd.DataFrame({ 'age'  : np.array([33, 51, 51, 34]), " + "'name' : pd.Categorical(['moon','jobs','gates','park'])})", context);
    assertEquals(ret.message().toString(), InterpreterResult.Code.SUCCESS, ret.code());
    //when
    ret = sql.interpret("select name, age from df2 where age < 40", context);
    //then
    assertEquals(new String(out.getOutputAt(0).toByteArray()), InterpreterResult.Code.SUCCESS, ret.code());
    assertEquals(new String(out.getOutputAt(0).toByteArray()), Type.TABLE, out.getOutputAt(0).getType());
    assertTrue(new String(out.getOutputAt(0).toByteArray()).indexOf("moon\t33") > 0);
    assertTrue(new String(out.getOutputAt(0).toByteArray()).indexOf("park\t34") > 0);
    assertEquals(InterpreterResult.Code.SUCCESS, sql.interpret("select case when name==\"aa\" then name else name end from df2", context).code());
}

Example 28 with InterpreterResult

use of org.apache.zeppelin.interpreter.InterpreterResult in project zeppelin by apache.

the class PythonInterpreterPandasSqlTest method setUp.

@Before
public void setUp() throws Exception {
    Properties p = new Properties();
    p.setProperty("zeppelin.python", "python");
    p.setProperty("zeppelin.python.maxResult", "100");
    intpGroup = new InterpreterGroup();
    python = new PythonInterpreter(p);
    python.setInterpreterGroup(intpGroup);
    python.open();
    sql = new PythonInterpreterPandasSql(p);
    sql.setInterpreterGroup(intpGroup);
    intpGroup.put("note", Arrays.asList(python, sql));
    out = new InterpreterOutput(this);
    context = new InterpreterContext("note", "id", null, "title", "text", new AuthenticationInfo(), new HashMap<String, Object>(), new GUI(), new AngularObjectRegistry(intpGroup.getId(), null), new LocalResourcePool("id"), new LinkedList<InterpreterContextRunner>(), out);
    // to make sure python is running.
    InterpreterResult ret = python.interpret("\n", context);
    assertEquals(ret.message().toString(), InterpreterResult.Code.SUCCESS, ret.code());
    sql.open();
}

Example 29 with InterpreterResult

use of org.apache.zeppelin.interpreter.InterpreterResult in project zeppelin by apache.

the class PythonInterpreterTest method testInterpret.

@Test
public void testInterpret() throws InterruptedException, IOException {
    InterpreterResult result = pythonInterpreter.interpret("print (\"hi\")", context);
    assertEquals(InterpreterResult.Code.SUCCESS, result.code());
}

Example 30 with InterpreterResult

use of org.apache.zeppelin.interpreter.InterpreterResult in project zeppelin by apache.

the class PythonInterpreterTest method testInterpretInvalidSyntax.

@Test
public void testInterpretInvalidSyntax() throws IOException {
    InterpreterResult result = pythonInterpreter.interpret("for x in range(0,3):  print (\"hi\")\n", context);
    assertEquals(InterpreterResult.Code.SUCCESS, result.code());
    assertTrue(new String(out.getOutputAt(0).toByteArray()).contains("hi\nhi\nhi"));
}