@Test
public void stripImports() {

    String script = "import org.apache.spark.sql._\n"
                      + "import org.apache.spark.sql.functions.{concat, lit, concat_ws,collect_list,split,size,col,when}\n"
                      +"import com.thinkbiganalytics.kylo.catalog._\n"
                      + "print 1";
    String actualScript = scalaImportManager.stripImports(script);
    assertThat(actualScript).isEqualToIgnoringWhitespace("print 1");
}