#!/usr/bin/env python3

import pyoasis
from pyoasis import OASIS
import numpy
from mpi4py import MPI

comm = MPI.COMM_WORLD

component_name = "receiver"

comp = pyoasis.Component(component_name, True, comm)
print(comp)

n_points = 16

partition = pyoasis.SerialPartition(n_points)
print(partition)

variable = pyoasis.Var("FRECVATM", partition, OASIS.IN)
print(variable)

comp.enddef()

date = int(0)
field = pyoasis.asarray(numpy.zeros(n_points))

variable.get(date, field)

expected_field = pyoasis.asarray(range(n_points))
epsilon = 1e-8
error = numpy.abs(field-expected_field).sum()
if error < epsilon:
    print("Data received successfully")

del comp