Added test for the getDataArray function