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