Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / tests / unit / inference_engine_tests / device_tests.cpp

diff --git a/inference-engine/tests/unit/inference_engine_tests/device_tests.cpp b/inference-engine/tests/unit/inference_engine_tests/device_tests.cpp
index c83d89a..368e8fd 100644 (file)
--- a/inference-engine/tests/unit/inference_engine_tests/device_tests.cpp
+++ b/inference-engine/tests/unit/inference_engine_tests/device_tests.cpp
@@ -1,4 +1,4 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
@@ -61,7 +61,8 @@ TEST_F(DeviceTests, returnsProperDeviceName) {
     ASSERT_STREQ(getDeviceName(TargetDevice::eMYRIAD), "MYRIAD");
     ASSERT_STREQ(getDeviceName(TargetDevice::eGNA), "GNA");
     ASSERT_STREQ(getDeviceName(TargetDevice::eHETERO), "HETERO");
+    ASSERT_STREQ(getDeviceName(TargetDevice::eKMB), "KMB");
     ASSERT_STREQ(getDeviceName(static_cast<TargetDevice>(-1)), "Unknown device");
     //off by one test - might not be enough
-    ASSERT_STREQ(getDeviceName(static_cast<TargetDevice>((uint8_t)TargetDevice::eHETERO + 1)), "Unknown device");
+    ASSERT_STREQ(getDeviceName(static_cast<TargetDevice>((uint8_t)TargetDevice::eKMB + 1)), "Unknown device");
 }