Add test from 16647

src/ndarray/ndarray.cc

@@ -249,7 +249,7 @@ NDArray NDArray::Reshape(const mxnet::TShape &shape) const {
  << "current shape.";
  } else {
  CHECK_GE(shape_.Size(), shape.Size())
- << "NDArray.Reshape: target shape size is larger current shape";
+ << "NDArray.Reshape: target shape size is larger than the current shape";
  }
  NDArray ret = this->Detach();
  // If the shape doesn't change, we can just return it now.

tests/python/unittest/test_ndarray.py

@@ -154,6 +154,22 @@ def test_ndarray_setitem():
  assert x.shape == trivial_shape
  assert same(x.asnumpy(), x_np)
 
+ # test https://github.com/apache/incubator-mxnet/issues/16647
+ dst = mx.nd.zeros((1, 3, 1)) # destination array
+ src = [1, 2, 3]
+ dst[0, :len(src), 0] = src
+ assert same(dst.asnumpy(), np.array([1, 2, 3], dtype=dst.dtype).reshape(dst.shape))
+
+ dst = mx.nd.zeros((1, 3, 1)) # destination array
+ src = [1, 2, 3]
+ dst[0, :len(src), 0] = mx.nd.array(src)
+ assert same(dst.asnumpy(), np.array([1, 2, 3], dtype=dst.dtype).reshape(dst.shape))
+
+ dst = mx.nd.zeros((1, 3, 1)) # destination array
+ src = [1, 2]
+ dst[0, :len(src), 0] = src
+ assert same(dst.asnumpy(), np.array([1, 2, 0], dtype=dst.dtype).reshape(dst.shape))
+
 
 @with_seed()
 def test_ndarray_elementwise():