ndarray: implement endof (#351)

dmlc · Dec 2, 2017 · 8a19651 · 8a19651
1 parent 09ee1f4
commit 8a19651
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/ndarray.jl b/src/ndarray.jl
@@ -295,6 +295,8 @@ end
 
 Base.first(x::NDArray) = _first(x)
 
+Base.endof(x::NDArray) = length(x)
+
 """
     slice(arr :: NDArray, start:stop)
 

diff --git a/test/unittest/ndarray.jl b/test/unittest/ndarray.jl
@@ -161,6 +161,13 @@ function test_first()
   end
 end  # function test_first
 
+function test_endof()
+  info("NDArray::endof")
+  let A = [1 2; 3 4; 5 6], x = mx.NDArray(A)
+    @test endof(A) == endof(x)
+  end
+end  # function test_endof
+
 function test_plus()
   dims   = rand_dims()
   t1, a1 = rand_tensors(dims)
@@ -789,6 +796,7 @@ end  # function test_size()
   test_slice()
   test_linear_idx()
   test_first()
+  test_endof()
   test_plus()
   test_minus()
   test_mul()