#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>

#include <xtensor/xarray.hpp>
#define FORCE_IMPORT_ARRAY
#include <xtensor-python/pyarray.hpp>

struct Quantity
{
    double q;
    
    Quantity(double q=0)
    : q(q)
    {
    }
};

PYBIND11_MODULE(test_array, test_array)
{
    xt::import_numpy();
    
    pybind11::class_<Quantity>(test_array, "Quantity")
        .def(pybind11::init<double>())
        .def_readwrite("q", &Quantity::q);
    
    test_array.def("create", []() { return xt::xarray<Quantity>{1.1, 2.2}; });
    test_array.def(
        "consume", [](xt::xarray<Quantity> const & array) {
            for(auto && item: array)
            {
                std::cout << item.q << "\n";
            }
        });
}

import numpy
import test_array

q1 = test_array.Quantity(42)
q2 = test_array.Quantity(3.14)
a = numpy.array([q1, q2])

print([x.q for x in a])

try:
    test_array.create()
except Exception as e:
    print(e)

try:
    test_array.consume(a)
except Exception as e:
    print(e)