Merge branch 'master' into seq_decomp

openvinotoolkit · Jul 19, 2023 · 23977e2 · 23977e2
2 parents cc3af4b + 9faa29d
commit 23977e2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/modules/java_api/src/test/java/org/intel/openvino/CoreTests.java b/modules/java_api/src/test/java/org/intel/openvino/CoreTests.java
@@ -25,6 +25,15 @@ public void testReadNetworkXmlOnly() {
         assertEquals("Batch size", 1, net.get_batch().get_length());
     }
 
+    @Test
+    public void testReadNetworkOnnx() {
+        Model net = core.read_model(modelOnnx);
+        PrePostProcessor p = new PrePostProcessor(net);
+        p.input().tensor().set_layout(new Layout("NCHW"));
+        p.build();
+        assertEquals("Batch size", 1, net.get_batch().get_length());
+    }
+
     @Test
     public void testReadModelIncorrectBinPath() {
         String exceptionMessage = "";

diff --git a/modules/java_api/src/test/java/org/intel/openvino/OVTest.java b/modules/java_api/src/test/java/org/intel/openvino/OVTest.java
@@ -11,6 +11,7 @@
 public class OVTest {
     String modelXml;
     String modelBin;
+    String modelOnnx;
     static String device;
 
     public OVTest() {
@@ -35,6 +36,14 @@ public OVTest() {
                                 "test_model",
                                 "test_model_fp32.bin")
                         .toString();
+
+        modelOnnx =
+                Paths.get(
+                                System.getProperty("MODELS_PATH"),
+                                "models",
+                                "test_model",
+                                "test_model.onnx")
+                        .toString();
     }
 
     @Rule