Updated display driver python test