saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding some unicode to the String inference test to make sure it doesn't fall prey to #2690.

Browse source
This commit is contained in:
Adam Pocock 2020-01-29 16:15:41 -05:00 committed by Changming Sun
parent f33759883c
commit 7bb5c357a8
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
java/src/test/java/ai/onnxruntime/InferenceTest.java
View file

@ -794,7 +794,7 @@ public class InferenceTest {
String inputName = session.getInputNames().iterator().next();
Map<String,OnnxTensor> container = new HashMap<>();
String[][] tensorIn = new String[][]{new String[] {"this", "is"}, new String[] {"identity", "test"}};
String[][] tensorIn = new String[][]{new String[] {"this", "is"}, new String[] {"identity", "test \u263A"}};
OnnxTensor ov = OnnxTensor.createTensor(env,tensorIn);
container.put(inputName,ov);
@ -809,14 +809,14 @@ public class InferenceTest {
assertEquals("this", labelOutput[0]);
assertEquals("is", labelOutput[1]);
assertEquals("identity", labelOutput[2]);
assertEquals("test", labelOutput[3]);
assertEquals("test \u263A", labelOutput[3]);
assertEquals(4, labelOutput.length);
OnnxValue.close(container);
container.clear();
}
String[] tensorInFlatArr = new String[]{"this", "is", "identity", "test"};
String[] tensorInFlatArr = new String[]{"this", "is", "identity", "test \u263A"};
ov = OnnxTensor.createTensor(env,tensorInFlatArr, new long[]{2,2});
container.put(inputName,ov);
@ -831,7 +831,7 @@ public class InferenceTest {
assertEquals("this", labelOutput[0]);
assertEquals("is", labelOutput[1]);
assertEquals("identity", labelOutput[2]);
assertEquals("test", labelOutput[3]);
assertEquals("test \u263A", labelOutput[3]);
assertEquals(4, labelOutput.length);
}
}