Add python grid test