# Copyright (C) 2018-2022 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

from openvino.tools.mo.front.tf.extractors.identity import tf_identity_ext
from unit_tests.utils.extractors import BaseExtractorsTestingClass


class IdentityExtractorTest(BaseExtractorsTestingClass):
    @classmethod
    def setUpClass(cls):
        cls.patcher = 'openvino.tools.mo.front.tf.extractors.identity.copy_shape_infer'

    def test_identity(self):
        self.expected = {}
        self.res = tf_identity_ext(pb=None)
        self.res["infer"](None)
        self.call_args = self.infer_mock.call_args
        self.expected_call_args = None
        self.compare()
