Add test for iterating over np.ndarray.

Author: chinganc

tests/unit_tests/test_nodes.py

@@ -2,7 +2,7 @@
 from opto.trace import node
 from opto.trace import operators as ops
 from opto.trace.utils import contain
-
+import numpy as np
 
 # Sum of str
 x = node("NodeX")
@@ -151,4 +151,11 @@ def fun(x):
 assert x.description == "[ParameterNode] x"
 
 x = node(1, trainable=True)
-assert x.description == "[ParameterNode] This is a ParameterNode in a computational graph."
+assert x.description == "[ParameterNode] This is a ParameterNode in a computational graph."
+
+
+# Test iterating numpy array
+x = node(np.array([1, 2, 3]))
+for i, v in enumerate(x):
+   assert isinstance(v, type(x))
+   assert v.data == x.data[i]