fix tests

Signed-off-by: fishbell <[email protected]>
songbell · Jan 2, 2024 · dd0fe59 · dd0fe59
1 parent e4f4f62
commit dd0fe59
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/plugins/hetero/tests/functional/hetero_tests.cpp b/src/plugins/hetero/tests/functional/hetero_tests.cpp
@@ -565,7 +565,7 @@ class MockPluginBase : public ov::IPlugin {
 class MockPluginReshape : public MockPluginBase {
 public:
     MockPluginReshape(const std::string& name)
-        : MockPluginBase(name, {"Parameter", "Result", "Add", "Constant", "Reshape"}, true) {}
+        : MockPluginBase(name, {"Parameter", "Result", "Add", "Constant", "Reshape", "Convert"}, true) {}
 
     const ov::Version& get_const_version() override {
         static const ov::Version version = {CI_BUILD_NUMBER, "openvino_mock_reshape_plugin"};
@@ -670,7 +670,7 @@ class MockPluginReshape : public MockPluginBase {
 class MockPluginSubtract : public MockPluginBase {
 public:
     MockPluginSubtract(const std::string& name)
-        : MockPluginBase(name, {"Parameter", "Result", "Add", "Constant", "Subtract"}) {}
+        : MockPluginBase(name, {"Parameter", "Result", "Add", "Constant", "Subtract", "Convert"}) {}
 
     const ov::Version& get_const_version() override {
         static const ov::Version version = {CI_BUILD_NUMBER, "openvino_mock_subtract_plugin"};

diff --git a/src/plugins/proxy/tests/proxy_tests.cpp b/src/plugins/proxy/tests/proxy_tests.cpp
@@ -512,7 +512,8 @@ void ov::proxy::tests::ProxyTests::register_plugin_support_reshape(ov::Core& cor
                                         const ov::AnyMap& properties) const override {
             OPENVINO_ASSERT(model);
 
-            std::unordered_set<std::string> supported_ops = {"Parameter", "Result", "Add", "Constant", "Reshape"};
+            std::unordered_set<std::string> supported_ops =
+                {"Parameter", "Result", "Add", "Constant", "Reshape", "Convert"};
 
             ov::SupportedOpsMap res;
             for (const auto& op : model->get_ordered_ops()) {
@@ -649,7 +650,8 @@ void ov::proxy::tests::ProxyTests::register_plugin_support_subtract(ov::Core& co
                                         const ov::AnyMap& properties) const override {
             OPENVINO_ASSERT(model);
 
-            std::unordered_set<std::string> supported_ops = {"Parameter", "Result", "Add", "Constant", "Subtract"};
+            std::unordered_set<std::string> supported_ops =
+                {"Parameter", "Result", "Add", "Constant", "Subtract", "Convert"};
 
             ov::SupportedOpsMap res;
             for (const auto& op : model->get_ordered_ops()) {