Type assertion to force evaluation of output NDArray

apache · Feb 26, 2020 · 556dbff · 556dbff
1 parent b8d3c02
commit 556dbff
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
@@ -535,6 +535,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()