[SPARK-2141] Adding one more doc test