Skip to content

Commit f45b35b

Browse files
RayWang-iatOceania2018
authored andcommitted
Update Math.Test.cs
1 parent 94edda5 commit f45b35b

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

test/TensorFlowNET.UnitTest/Numpy/Math.Test.cs

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -66,16 +66,19 @@ public void power()
6666
Assert.AreEqual(y, new[] { 0, 1, 8, 27, 64, 125 });
6767
}
6868
[TestMethod]
69-
public void maximum()
69+
public void maximum()
7070
{
7171
var x1 = new NDArray(new[,] { { 1, 2, 3 }, { 4, 5.1, 6 } });
7272
var x2 = new NDArray(new[,] { { 3, 2, 1 }, { 6, 5.1, 4 } });
73-
var y = np.maximum(x1,x2);
73+
var y0 = np.maximum(x1,x2);
7474
var y1 = np.maximum(x1, x2, axis: 0);
75-
var y2 = new NDArray(new[,] { { 3, 2, 3 }, { 6, 5.1, 6 } });
76-
var y3 = new NDArray(new[] { 6, 5.1, 6 });
77-
Assert.AreEqual(y, y2);
78-
Assert.AreEqual(y1, y3);
75+
var y2 = np.maximum(x1, x2, axis: 1);
76+
var y3 = new NDArray(new[,] { { 3, 2, 3 }, { 6, 5.1, 6 } });
77+
var y4 = new NDArray(new[] { 6, 5.1, 6 });
78+
var y5 = new NDArray(new[] { 3.0, 6 });
79+
Assert.AreEqual(y0, y3);
80+
Assert.AreEqual(y1, y4);
81+
Assert.AreEqual(y2, y5);
7982
}
8083
}
8184
}

0 commit comments

Comments
 (0)