from mpi4py import MPI
import numpy as np
import healpy as hp

mpi = False
rank = MPI.COMM_WORLD.Get_rank()

nside = 32
npix = hp.nside2npix(nside)

nrings = 4 * nside - 1 # four missing pixels

import libsharp

n_mpi_processes = MPI.COMM_WORLD.Get_size()

assert n_mpi_processes == 1 # run only with 1 process

ring_indices = np.arange(nrings, dtype=np.int32)

assert len(ring_indices) == nrings
if not mpi:
    ring_indices = None
grid = libsharp.healpix_grid(nside, rings=ring_indices)

local_npix = grid.local_size()
print("npix", npix, "local_npix", local_npix)