83 lines
2.5 KiB
Java
83 lines
2.5 KiB
Java
import static org.junit.Assert.*;
|
|
|
|
import org.intel.openvino.*;
|
|
import org.junit.Test;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
public class IECoreTests extends IETest {
|
|
IECore core = new IECore();
|
|
|
|
@Test
|
|
public void testReadNetwork() {
|
|
CNNNetwork net = core.ReadNetwork(modelXml, modelBin);
|
|
assertEquals("Network name", "test_model", net.getName());
|
|
}
|
|
|
|
@Test
|
|
public void testReadNetworkXmlOnly() {
|
|
CNNNetwork net = core.ReadNetwork(modelXml);
|
|
assertEquals("Batch size", 1, net.getBatchSize());
|
|
}
|
|
|
|
@Test
|
|
public void testReadNetworkIncorrectXmlPath() {
|
|
String exceptionMessage = "";
|
|
try {
|
|
CNNNetwork net = core.ReadNetwork("model.xml", modelBin);
|
|
} catch (Exception e) {
|
|
exceptionMessage = e.getMessage();
|
|
}
|
|
assertTrue(exceptionMessage.contains("Model file model.xml cannot be opened!"));
|
|
}
|
|
|
|
@Test
|
|
public void testReadNetworkIncorrectBinPath() {
|
|
String exceptionMessage = "";
|
|
try {
|
|
CNNNetwork net = core.ReadNetwork(modelXml, "model.bin");
|
|
} catch (Exception e) {
|
|
exceptionMessage = e.getMessage();
|
|
}
|
|
assertTrue(exceptionMessage.contains("Weights file model.bin cannot be opened!"));
|
|
}
|
|
|
|
@Test
|
|
public void testLoadNetwork() {
|
|
CNNNetwork net = core.ReadNetwork(modelXml, modelBin);
|
|
ExecutableNetwork executableNetwork = core.LoadNetwork(net, device);
|
|
|
|
assertTrue(executableNetwork instanceof ExecutableNetwork);
|
|
}
|
|
|
|
@Test
|
|
public void testLoadNetworDeviceConfig() {
|
|
CNNNetwork net = core.ReadNetwork(modelXml, modelBin);
|
|
|
|
Map<String, String> testMap = new HashMap<String, String>();
|
|
|
|
// When specifying key values as raw strings, omit the KEY_ prefix
|
|
testMap.put("CPU_BIND_THREAD", "YES");
|
|
testMap.put("CPU_THREADS_NUM", "1");
|
|
|
|
ExecutableNetwork executableNetwork = core.LoadNetwork(net, device, testMap);
|
|
|
|
assertTrue(executableNetwork instanceof ExecutableNetwork);
|
|
}
|
|
|
|
@Test
|
|
public void testLoadNetworkWrongDevice() {
|
|
String exceptionMessage = "";
|
|
CNNNetwork net = core.ReadNetwork(modelXml, modelBin);
|
|
try {
|
|
core.LoadNetwork(net, "DEVISE");
|
|
} catch (Exception e) {
|
|
exceptionMessage = e.getMessage();
|
|
}
|
|
assertTrue(
|
|
exceptionMessage.contains(
|
|
"Device with \"DEVISE\" name is not registered in the InferenceEngine"));
|
|
}
|
|
}
|