Searching in python/pyspark/sql/tests/plot