Function validate_npy_header

tests/AssetsLibrary.cpp:521–555

namespace detail
{
inline void
validate_npy_header(std::ifstream &stream, const std::string &expect_typestr, const TensorShape &expect_shape)
{
    ARM_COMPUTE_UNUSED(expect_typestr);
    ARM_COMPUTE_UNUSED(expect_shape);

    std::string header_s = npy::read_header(stream);

    // Parse header
    npy::header_t header = npy::parse_header(header_s);

    std::vector<unsigned long> shape         = header.shape;
    bool                       fortran_order = header.fortran_order;
    std::string                typestr       = header.dtype.str();

    // Check if the typestring matches the given one
    ARM_COMPUTE_ERROR_ON_MSG(typestr != expect_typestr, "Typestrings mismatch");

    // Validate tensor shape
    ARM_COMPUTE_ERROR_ON_MSG(shape.size() != expect_shape.num_dimensions(), "Tensor ranks mismatch");
    if (fortran_order)
    {
        for (size_t i = 0; i < shape.size(); ++i)
        {
            ARM_COMPUTE_ERROR_ON_MSG(expect_shape[i] != shape[i], "Tensor dimensions mismatch");
        }
    }
    else
    {
        for (size_t i = 0; i < shape.size(); ++i)
        {
            ARM_COMPUTE_ERROR_ON_MSG(expect_shape[i] != shape[shape.size() - i - 1], "Tensor dimensions mismatch");
        }
    }
}
} // namespace detail
} // namespace test
} // namespace arm_compute