Change to use iterator so that we can check the result of execution.

Author: bgogul

Tests/TensorFlowTests/LazyTensorEvaluationTests.swift

@@ -87,14 +87,32 @@ final class LazyTensorEvaluationTests: XCTestCase {
     }
 
     func testNoOutputOperations() {
-        let a = StringTensor("Hello ")
-        let b = StringTensor("World")
-        let c = Raw.add(a, b)
-        // c is not materialized yet.
-        XCTAssertFalse(isMaterialized(c.handle.handle))
-        Raw.printV2(c)
-        // c is materialized now as printV2 would be executed.
-        XCTAssertTrue(isMaterialized(c.handle.handle))
+        let elements: Tensor<Int32> = [0, 1, 2]
+        // let elements2: Tensor<Int32> = [10, 11, 12]
+        let outputTypes = [Int32.tensorFlowDataType]
+        let outputShapes: [TensorShape?] = [nil]
+        let dataset: VariantHandle = Raw.tensorSliceDataset(
+            components: [elements],
+            outputShapes: [nil]
+        )
+        let iterator: ResourceHandle = Raw.iteratorV2(sharedName: "blah",
+            container: "earth", outputTypes: outputTypes, outputShapes: outputShapes
+        )
+        // `dataset` and `iterator` should not be materialized yet.
+        XCTAssertFalse(isMaterialized(dataset.handle))
+        XCTAssertFalse(isMaterialized(iterator.handle))
+
+        Raw.makeIterator(dataset: dataset, iterator: iterator)
+        // `dataset` and `iterator` should be materialized now as
+        // makeIterator executes.
+        XCTAssertTrue(isMaterialized(dataset.handle))
+        XCTAssertTrue(isMaterialized(iterator.handle))
+
+        // The following won't work unless the makeIterator runs.
+        let next: TensorHandle<Int32> = Raw.iteratorGetNext(
+            iterator: iterator, outputShapes: outputShapes
+        )
+        XCTAssertEqual(Tensor(handle: next).scalarized(), 0)
     }
 
     private func isMaterialized<T: TensorFlowScalar>(_ input: Tensor<T>) -> Bool {