Type assertion to force evaluation of output NDArray

apache · Mar 3, 2020 · eb09cf1 · eb09cf1
1 parent 671ee0a
commit eb09cf1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -562,6 +562,8 @@ def check_rnn():
             assert out.shape[1] == 4
             assert out.shape[2] == 1
 
+            assert type(out[0, 0, 0].asscalar()).__name__ == 'float32'
+
     check_gluon_embedding()
     check_fully_connected()
     check_dense()