diff --git a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs index 25ffe217..47d53824 100644 --- a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs +++ b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs @@ -45,5 +45,21 @@ namespace TensorFlowNET.UnitTest.Dataset value += 2; } } + + [TestMethod] + public void FromTensorSlices() + { + var X = tf.constant(new[] { 2013, 2014, 2015, 2016, 2017 }); + var Y = tf.constant(new[] { 12000, 14000, 15000, 16500, 17500 }); + + var dataset = tf.data.Dataset.from_tensor_slices(X, Y); + int n = 0; + foreach (var (item_x, item_y) in dataset) + { + print($"x:{item_x.numpy()},y:{item_y.numpy()}"); + n += 1; + } + Assert.AreEqual(5, n); + } } } diff --git a/test/TensorFlowNET.UnitTest/TF_API/tfdata_test.cs b/test/TensorFlowNET.UnitTest/TF_API/tfdata_test.cs deleted file mode 100644 index e8147b6e..00000000 --- a/test/TensorFlowNET.UnitTest/TF_API/tfdata_test.cs +++ /dev/null @@ -1,27 +0,0 @@ -using Microsoft.VisualStudio.TestTools.UnitTesting; -using Tensorflow; -using static Tensorflow.Binding; - -namespace TensorFlowNET.UnitTest.TF_API -{ - [TestClass] - public class tfdata_test - { - [TestMethod] - public void CreateFromTensor() - { - var X = tf.constant(new[] { 2013, 2014, 2015, 2016, 2017 }); - var Y = tf.constant(new[] { 12000, 14000, 15000, 16500, 17500 }); - - var dataset = tf.data.Dataset.from_tensor_slices(X, Y); - int n = 0; - foreach (var (item_x, item_y) in dataset) - { - print($"x:{item_x.numpy()},y:{item_y.numpy()}"); - n += 1; - } - Assert.AreEqual(5, n); - } - - } -}