test meshgrid_same_ndim

Oceania2018 · Oceania2018 · commit f32314e85b9b · 2021-07-15T21:01:54.000-05:00
diff --git a/test/TensorFlowNET.UnitTest/Numpy/Array.Creation.Test.cs b/test/TensorFlowNET.UnitTest/Numpy/Array.Creation.Test.cs
@@ -80,5 +80,13 @@ public void meshgrid()
             Assert.IsTrue(Equal(new double[] { 0, 1 }, yv.ToArray<double>()));
             AssetSequenceEqual(new long[] { 2, 1 }, yv.shape.dims);
         }
+
+        [TestMethod]
+        public void meshgrid_same_ndim()
+        {
+            var (a, b) = np.meshgrid(np.arange(3), np.arange(3));
+            AssetSequenceEqual(a.ToArray<int>(), new int[] { 0, 1, 2, 0, 1, 2, 0, 1, 2 });
+            AssetSequenceEqual(b.ToArray<int>(), new int[] { 0, 0, 0, 1, 1, 1, 2, 2, 2 });
+        }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -80,5 +80,13 @@ public void meshgrid()`
`80`	`80`	`Assert.IsTrue(Equal(new double[] { 0, 1 }, yv.ToArray<double>()));`
`81`	`81`	`AssetSequenceEqual(new long[] { 2, 1 }, yv.shape.dims);`
`82`	`82`	`}`
	`83`	`+`
	`84`	`+ [TestMethod]`
	`85`	`+ public void meshgrid_same_ndim()`
	`86`	`+ {`
	`87`	`+ var (a, b) = np.meshgrid(np.arange(3), np.arange(3));`
	`88`	`+ AssetSequenceEqual(a.ToArray<int>(), new int[] { 0, 1, 2, 0, 1, 2, 0, 1, 2 });`
	`89`	`+ AssetSequenceEqual(b.ToArray<int>(), new int[] { 0, 0, 0, 1, 1, 1, 2, 2, 2 });`
	`90`	`+ }`
`83`	`91`	`}`
`84`	`92`	`}`