Added test for dimensionality checks