[MXNET-1408] Adding test to verify Large Tensor Support for ravel and unravel

Author: Rohit Kumar Srivastava

3rdparty/dmlc-core

@@ -279,6 +279,19 @@ def test_diag():
     assert_almost_equal(r.asnumpy(), np.diag(a_np, k=k))
 
 
+def test_ravel_and_unravel():
+    idxs_2d = [[LARGE_X-1,LARGE_X-100,6],[SMALL_Y-1,SMALL_Y-10,1]]
+    idx = mx.nd.array(1)
+    def test_ravel_multi_index():
+        idx = mx.nd.ravel_multi_index(mx.nd.array(idxs_2d, dtype=np.int64), shape=(LARGE_X,SMALL_Y))
+        idx_numpy = np.ravel_multi_index(idxs_2d, (LARGE_X,SMALL_Y))
+        assert np.sum(1 for i in range(idx.size) if idx[i] == idx_numpy[i]) == 3
+
+    def test_unravel_index():
+        original_2d_idxs = mx.nd.unravel_index(mx.nd.array(idx, dtype=np.int64), shape=(LARGE_X,SMALL_Y))
+        assert (original_2d_idxs.asnumpy() == np.array(idxs_2d)).all()
+
+
 if __name__ == '__main__':
     import nose
     nose.runmodule()