Adding failing test with dense_array_equal due to shape mismatch