@Test
public void testConvertToArrowVectors() {

    INDArray matrix = Nd4j.linspace(1,4,4).reshape(2,2);
    val vectors = ArrowConverter.convertToArrowVector(matrix,Arrays.asList("test","test2"), ColumnType.Double,bufferAllocator);
    assertEquals(matrix.rows(),vectors.size());
    INDArray vector = Nd4j.linspace(1,4,4);
    val vectors2 = ArrowConverter.convertToArrowVector(vector,Arrays.asList("test"), ColumnType.Double,bufferAllocator);
    assertEquals(1,vectors2.size());
    assertEquals(matrix.length(),vectors2.get(0).getValueCount());
}