IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor of tests (#14544)

Browse Source
This commit is contained in:
Tomasz Dołbniak 2022-12-12 03:17:14 +01:00 committed by GitHub
parent 3c89da1838
commit 11d9b01adb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 150 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

190
src/core/tests/opset.cpp
View File

@ -18,165 +18,55 @@
#include "openvino/opsets/opset8.hpp" #include "openvino/opsets/opset8.hpp"
#include "openvino/opsets/opset9.hpp" #include "openvino/opsets/opset9.hpp"
TEST(opset, opset1) { struct OpsetTestParams {
auto op = std::make_shared<ov::opset1::Parameter>(); using OpsetGetterFunction = std::function<const ov::OpSet&()>;
OpsetTestParams(const OpsetGetterFunction& opset_getter_, const uint32_t expected_ops_count_)
: opset_getter{opset_getter_},
expected_ops_count{expected_ops_count_} {}
OpsetGetterFunction opset_getter;
uint32_t expected_ops_count;
};
class OpsetTests : public testing::TestWithParam<OpsetTestParams> {};
struct OpsetTestNameGenerator {
std::string operator()(const testing::TestParamInfo<OpsetTestParams>& info) const {
return "opset" + std::to_string(info.index + 1);
}
};
TEST_P(OpsetTests, create_parameter) {
const auto& params = GetParam();
const auto op = std::unique_ptr<ov::Node>(params.opset_getter().create("Parameter"));
ASSERT_NE(nullptr, op); ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op)); EXPECT_TRUE(ov::op::util::is_parameter(op.get()));
} }
TEST(opset, opset1_dump) { TEST_P(OpsetTests, opset_dump) {
const auto& opset = ov::get_opset1(); const auto& params = GetParam();
std::cout << "All opset1 operations: "; const auto& opset = params.opset_getter();
std::cout << "All opset operations: ";
for (const auto& t : opset.get_types_info()) { for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " "; std::cout << t.name << " ";
} }
std::cout << std::endl; std::cout << std::endl;
ASSERT_EQ(110, opset.get_types_info().size()); ASSERT_EQ(params.expected_ops_count, opset.get_types_info().size());
} }
TEST(opset, opset2) { INSTANTIATE_TEST_SUITE_P(opset,
auto op = std::make_shared<ov::opset2::Parameter>(); OpsetTests,
ASSERT_NE(nullptr, op); testing::Values(OpsetTestParams{ov::get_opset1, 110},
EXPECT_TRUE(ov::op::util::is_parameter(op)); OpsetTestParams{ov::get_opset2, 112},
} OpsetTestParams{ov::get_opset3, 127},
OpsetTestParams{ov::get_opset4, 137},
TEST(opset, opset2_dump) { OpsetTestParams{ov::get_opset5, 145},
const auto& opset = ov::get_opset2(); OpsetTestParams{ov::get_opset6, 152},
std::cout << "All opset2 operations: "; OpsetTestParams{ov::get_opset7, 156},
for (const auto& t : opset.get_types_info()) { OpsetTestParams{ov::get_opset8, 167},
std::cout << t.name << " "; OpsetTestParams{ov::get_opset9, 173},
} OpsetTestParams{ov::get_opset10, 177}),
std::cout << std::endl; OpsetTestNameGenerator{});
ASSERT_EQ(112, opset.get_types_info().size());
}
TEST(opset, opset3) {
auto op = std::make_shared<ov::opset3::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset3_dump) {
const auto& opset = ov::get_opset3();
std::cout << "All opset3 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(127, opset.get_types_info().size());
}
TEST(opset, opset4) {
auto op = std::make_shared<ov::opset4::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset4_dump) {
const auto& opset = ov::get_opset4();
std::cout << "All opset4 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(137, opset.get_types_info().size());
}
TEST(opset, opset5) {
auto op = std::make_shared<ov::opset5::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset5_dump) {
const auto& opset = ov::get_opset5();
std::cout << "All opset5 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(145, opset.get_types_info().size());
}
TEST(opset, opset6) {
auto op = std::make_shared<ov::opset6::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset6_dump) {
const auto& opset = ov::get_opset6();
std::cout << "All opset6 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(152, opset.get_types_info().size());
}
TEST(opset, opset7) {
auto op = std::make_shared<ov::opset7::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset7_dump) {
const auto& opset = ov::get_opset7();
std::cout << "All opset7 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(156, opset.get_types_info().size());
}
TEST(opset, opset8) {
auto op = std::make_shared<ov::opset8::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset8_dump) {
const auto& opset = ov::get_opset8();
std::cout << "All opset8 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(167, opset.get_types_info().size());
}
TEST(opset, opset9) {
auto op = std::make_shared<ov::opset9::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset9_dump) {
const auto& opset = ov::get_opset9();
std::cout << "All opset9 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(173, opset.get_types_info().size());
}
TEST(opset, opset10) {
auto op = std::make_shared<ov::opset10::Parameter>();
ASSERT_NE(nullptr, op);
EXPECT_TRUE(ov::op::util::is_parameter(op));
}
TEST(opset, opset10_dump) {
const auto& opset = ov::get_opset10();
std::cout << "All opset10 operations: ";
for (const auto& t : opset.get_types_info()) {
std::cout << t.name << " ";
}
std::cout << std::endl;
ASSERT_EQ(177, opset.get_types_info().size());
}
class MyOpOld : public ov::op::Op { class MyOpOld : public ov::op::Op {
public: public: