python option to set up grid

- TODO: add test to compare fortran and python
  outputs