You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by wa...@apache.org on 2019/11/13 01:52:42 UTC
[singa] branch master updated: [SINGA-495] Improve Java tensor test
case format
This is an automated email from the ASF dual-hosted git repository.
wangwei pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/singa.git
The following commit(s) were added to refs/heads/master by this push:
new 3e0571f [SINGA-495] Improve Java tensor test case format
new a1ad4d8 Merge pull request #551 from Wentong-DST/singa-495
3e0571f is described below
commit 3e0571f2398379f1f2d02d1586a2771c8eb15744
Author: Wentong <we...@grabtaxi.com>
AuthorDate: Mon Oct 21 12:03:08 2019 +0800
[SINGA-495] Improve Java tensor test case format
---
.../src/test/java/org/apache/singa/TestTensor.java | 59 ++++++++++------------
1 file changed, 28 insertions(+), 31 deletions(-)
diff --git a/java/src/test/java/org/apache/singa/TestTensor.java b/java/src/test/java/org/apache/singa/TestTensor.java
index c66f41a..7ef294e 100644
--- a/java/src/test/java/org/apache/singa/TestTensor.java
+++ b/java/src/test/java/org/apache/singa/TestTensor.java
@@ -19,41 +19,38 @@
package test;
+import junit.framework.*;
import org.apache.singa.swig.*;
-
-import junit.framework.*;
import static org.junit.Assert.*;
public class TestTensor extends TestCase {
- protected void setUp(){
- System.loadLibrary("singa_wrap");
- }
- public void testTensorFunc() {
- Shape s = new Shape(2);
- s.set(0, 2);
- s.set(1, 3);
-
- Tensor t1 = new Tensor(s);
- t1.SetFloatValue(0.1f);
- Tensor t2 = singa_wrap.Square(t1);
- float[] data = new float[6];
-
- t2.GetFloatValue(data, 6);
- for(int i = 0; i < 6; i++)
- assertEquals(data[i], 0.01, 1e-4);
-
- for (int i =0; i< 6; i++)
- data[i] = i * 1.0f;
- Tensor t3 = new Tensor(s);
- t3.CopyFloatDataFromHostPtr(data, 6);
-
-
- t3.GetFloatValue(data, 6);
- for(int i = 0; i < 6; i++)
- assertEquals(data[i], i * 1.0f, 1e-4);
-
- }
-
+ protected void setUp() {
+ System.loadLibrary("singa_wrap");
+ }
+
+ public void testTensorFunc() {
+ Shape s = new Shape(2);
+ s.set(0, 2);
+ s.set(1, 3);
+
+ Tensor t1 = new Tensor(s);
+ t1.SetFloatValue(0.1f);
+ Tensor t2 = singa_wrap.Square(t1);
+ float[] data = new float[6];
+
+ t2.GetFloatValue(data, 6);
+ for (int i = 0; i < 6; i++)
+ assertEquals(data[i], 0.01, 1e-4);
+
+ for (int i = 0; i < 6; i++)
+ data[i] = i * 1.0f;
+ Tensor t3 = new Tensor(s);
+ t3.CopyFloatDataFromHostPtr(data, 6);
+
+ t3.GetFloatValue(data, 6);
+ for (int i = 0; i < 6; i++)
+ assertEquals(data[i], i * 1.0f, 1e-4);
+ }
}