inference-engine/tests/unit/mocks/mock_iformat_parser.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 /**
   6  * \brief mock file for header file for IFormatParser
   7  * \file mock_iformat_parser.hpp
   8  */
   9 #pragma once
  10
  11 #include "ie_icnn_network.hpp"
  12 #include <gmock/gmock-generated-function-mockers.h>
  13 #include <inference_engine/ie_cnn_net_reader_impl.h>
  14 #include <inference_engine/parsers.h>
  15 #include "pugixml.hpp"
  16
  17 struct MockIFormatParser : public InferenceEngine::details::IFormatParser {
  18     public:
  19     MOCK_METHOD1(Parse, InferenceEngine::details::CNNNetworkImplPtr(pugi::xml_node &));
  20
  21     MOCK_METHOD1(SetWeights, void(const InferenceEngine::TBlob<uint8_t>::Ptr &));
  22 };
  23