Add Python tests