inference-engine/tests/unit/inference_engine_tests/inference_engine_test.cpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <gtest/gtest.h>
   6
   7 #include "inference_engine.hpp"
   8
   9 using namespace std;
  10 using namespace testing;
  11 using namespace InferenceEngine;
  12
  13 class InferenceEngineTests : public ::testing::Test {
  14 public:
  15         InferenceEngineTests(): output(Precision::FP32, C)
  16         {
  17         }
  18
  19 protected:
  20     InferenceEngine::TBlob<float> output;
  21     vector<unsigned> results;
  22     virtual void TearDown() override{
  23     }
  24
  25     virtual void SetUp() override{
  26         output.Resize({10, 1}, Layout::NC);
  27         output.set({ 0.3f, 0.1f, 0.01f, 0.9f, 0.99f, 0.12f, 0.001f, 0, 0.999f, 0.0000001f });
  28     }
  29
  30     InferenceEngine::TBlob<float>::Ptr getCopiedTBlob(InferenceEngine::SizeVector size) {
  31         InferenceEngine::TBlob<float>::Ptr blob(new InferenceEngine::TBlob<float>(Precision::FP32,
  32                                                                                   TensorDesc::getLayoutByDims(size), size));
  33         blob->allocate();
  34         const size_t arr_size = 4;
  35         uint8_t data[arr_size] = { 1, 2, 3, 4 };
  36         InferenceEngine::copyFromRGB8(&data[0], arr_size, blob.get());
  37         return blob;
  38     }
  39 };
  40
  41 TEST_F(InferenceEngineTests, checkZeroInput) {
  42     InferenceEngine::TBlob<float> output(Precision::FP32, C);
  43     output.set({});
  44     EXPECT_THROW(InferenceEngine::TopResults(5, output, results), InferenceEngine::details::InferenceEngineException);
  45 }
  46
  47 TEST_F(InferenceEngineTests, testInsertSort) {
  48
  49     InferenceEngine::TopResults(5, output, results);
  50     ASSERT_EQ(5, results.size());
  51     ASSERT_EQ(8, results[0]);
  52     ASSERT_EQ(4, results[1]);
  53     ASSERT_EQ(3, results[2]);
  54     ASSERT_EQ(0, results[3]);
  55     ASSERT_EQ(5, results[4]);
  56 }
  57
  58 TEST_F(InferenceEngineTests, testInsertSortOverDraft) {
  59
  60     InferenceEngine::TopResults(15, output, results);
  61     ASSERT_EQ(10, results.size());
  62     ASSERT_EQ(8, results[0]);
  63     ASSERT_EQ(4, results[1]);
  64     ASSERT_EQ(3, results[2]);
  65     ASSERT_EQ(0, results[3]);
  66     ASSERT_EQ(5, results[4]);
  67     ASSERT_EQ(1, results[5]);
  68     ASSERT_EQ(2, results[6]);
  69     ASSERT_EQ(6, results[7]);
  70     ASSERT_EQ(9, results[8]);
  71     ASSERT_EQ(7, results[9]);
  72 }
  73
  74 TEST_F(InferenceEngineTests, testThrowsOnCopyToBadBlob) {
  75     ASSERT_THROW(getCopiedTBlob({ 1, 1, 1 }), InferenceEngine::details::InferenceEngineException);
  76 }
  77
  78 TEST_F(InferenceEngineTests, testThrowsOnCopyToBlobWithBadSize) {
  79     ASSERT_THROW(getCopiedTBlob({ 1, 1, 1, 1 }), InferenceEngine::details::InferenceEngineException);
  80 }
  81
  82 TEST_F(InferenceEngineTests, canCopyToProperBlob) {
  83     auto blob = getCopiedTBlob({ 1, 1, 1, 4 });
  84     ASSERT_EQ(blob->data()[blob->size() - 1], 4);
  85 }