[SPARK-2141] updating doctest