Moved MO unit test files to a separate directory (#5312)

* Removed test-generator from all MO requirement files except the dev one

* Moved all MO unit tests files to a separate directory

* Added __init__.py files to the tests directory. Fixed importing paths for some unit tests

* Fixed imports in all unit tests. Moved all unit test related files from the MO code to the dedicated directory

* Renamed directory with unit test utils

* Updated imports in unit tests

model-optimizer/requirements.txt

@@ -5,7 +5,6 @@ networkx>=1.11
 numpy>=1.14.0,<1.19.0
 protobuf>=3.6.1
 onnx>=1.1.2
-test-generator==0.1.1
 defusedxml>=0.5.0
 urllib3>=1.25.9
 requests>=2.20.0

model-optimizer/requirements_caffe.txt

@@ -1,6 +1,5 @@
 networkx>=1.11
 numpy>=1.14.0
 protobuf>=3.6.1
-test-generator==0.1.1
 defusedxml>=0.5.0
 requests>=2.20.0

model-optimizer/requirements_kaldi.txt

@@ -1,5 +1,4 @@
 networkx>=1.11
 numpy>=1.14.0
-test-generator==0.1.1
 defusedxml>=0.5.0
 requests>=2.20.0

model-optimizer/requirements_mxnet.txt

@@ -1,7 +1,6 @@
 mxnet>=1.0.0,<=1.7.0
 networkx>=1.11
 numpy>=1.14.0
-test-generator==0.1.1
 defusedxml>=0.5.0
 urllib3>=1.25.9
 requests>=2.20.0

model-optimizer/requirements_onnx.txt

@@ -1,6 +1,5 @@
 onnx>=1.1.2
 networkx>=1.11
 numpy>=1.14.0
-test-generator==0.1.1
 defusedxml>=0.5.0
 requests>=2.20.0

model-optimizer/requirements_tf.txt

@@ -2,6 +2,5 @@ tensorflow>=1.15.2,<2.0; python_version < "3.8"
 tensorflow>=2.2; python_version >= "3.8"
 networkx>=1.11
 numpy>=1.14.0,<1.19.0
-test-generator==0.1.1
 defusedxml>=0.5.0
 requests>=2.20.0

model-optimizer/requirements_tf2.txt

@@ -1,6 +1,5 @@
 tensorflow>=2.2
 networkx>=1.11
 numpy>=1.14.0
-test-generator==0.1.1
 defusedxml>=0.5.0
 requests>=2.20.0

model-optimizer/telemetry/utils/sender_test.py

@@ -1,68 +0,0 @@
-# Copyright (C) 2018-2021 Intel Corporation
-# SPDX-License-Identifier: Apache-2.0
-
-import unittest
-import time
-
-from telemetry.utils.sender import TelemetrySender
-
-
-class FakeTelemetryBackend:
-    def send(self, param):
-        pass
-
-
-class FakeTelemetryBackendWithSleep:
-    def send(self, param):
-        time.sleep(1)
-
-
-class TelemetrySenderStress(unittest.TestCase):
-    def test_stress(self):
-        """
-        Stress tests to schedule a lot of threads which works super fast (do nothing) with sending messages."
-        """
-        tm = TelemetrySender()
-        fake_backend = FakeTelemetryBackend()
-        for _ in range(1000000):
-            tm.send(fake_backend, None)
-
-    def test_check_shutdown(self):
-        """
-        Stress test to schedule many threads taking 1 second and then ask to force shutdown. Make sure that the elapsed
-        time is small.
-        """
-        tm = TelemetrySender()
-        fake_backend = FakeTelemetryBackendWithSleep()
-        # schedule many requests which just wait 1 second
-        for _ in range(100000):
-            tm.send(fake_backend, None)
-
-        start_time = time.time()
-        # ask to shutdown with timeout of 1 second
-        tm.force_shutdown(1)
-        while len(tm.executor._threads):
-            pass
-        # check that no more than 3 seconds spent
-        self.assertTrue(time.time() - start_time < 3)
-
-    def test_check_shutdown_negative(self):
-        """
-        Test to check that without forcing shutdown total execution time is expected.
-        """
-        tm = TelemetrySender(1)  # only one worker thread
-        fake_backend = FakeTelemetryBackendWithSleep()
-        start_time = time.time()
-        # schedule 5 requests which totally should work more than 4 seconds
-        for _ in range(5):
-            tm.send(fake_backend, None)
-
-        try:
-            # wait until all threads finish their work. We use internal ThreadPoolExecutor attribute _work_queue to make
-            # sure that all workers completed their work, so the whole code is wrapped to try/except to avoid exceptions
-            # if internal implementation is changed in the future
-            while tm.executor._work_queue.qsize():
-                pass
-            self.assertTrue(time.time() - start_time > 4.0)
-        except:
-            pass

model-optimizer/unit_tests/extensions/analysis/Iterator_get_next_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.analysis.inputs import InputsAnalysis
 from mo.front.common.partial_infer.utils import int64_array
-from mo.utils.unittest.graph import build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph_with_edge_attrs
 
 
 class IteratorGetNextAnalysisTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/ClampNormalizer_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.back.ClampNormalizer import ClampNormalizer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, result, connect
+from unit_tests.utils.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, result, connect
 
 
 class AttributedClampNormalizerTests(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/ConvolutionNormalizer_test.py

@@ -11,7 +11,7 @@ from extensions.ops.fakequantize import FakeQuantize
 from mo.front.common.partial_infer.utils import int64_array
 from mo.ops.reshape import Reshape
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, regular_op_with_empty_data, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, regular_op_with_empty_data, \
     valued_const_with_data, connect
 
 

model-optimizer/unit_tests/extensions/back/CutMemory_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.back.CutMemory import CutMemoryInput, CutMemoryOutput
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class CutMemoryTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/FakeOutputResolver_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.back.FakeOutputResolver import FakeOutputResolver
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_empty_data, connect, empty_data, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_empty_data, connect, empty_data, \
     valued_const_with_data
 
 

model-optimizer/unit_tests/extensions/back/FuseTransposesSequence_test.py

@@ -2,13 +2,12 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import unittest
-from argparse import Namespace
 
 import numpy as np
 
 from extensions.back.FuseTransposesSequence import FuseTransposesSequence
-from mo.middle.passes.eliminate_test import build_graph
 from mo.utils.ir_engine.compare_graphs import compare_graphs
+from unit_tests.utils.graph import build_graph
 
 # The dictionary with nodes attributes used to build various graphs. A key is the name of the node and the value is the
 # dictionary with node attributes.

model-optimizer/unit_tests/extensions/back/InterpolateReshape_test.py

@@ -2,13 +2,12 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import unittest
-from argparse import Namespace
 
 import numpy as np
 
 from extensions.back.InterpolateReshape import InterpolateReshapeWA, InterpolateConcat
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
     connect_data
 
 nodes = {

model-optimizer/unit_tests/extensions/back/LayoutChangeForGatherND_test.py

@@ -8,7 +8,7 @@ import numpy as np
 from extensions.back.LayoutChangeForGatherND import LayoutChangeForGatherND
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'placeholder_1': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/back/MarkNodesWithShapeValues_test.py

@@ -9,9 +9,8 @@ from extensions.back.MarkNodesWithShapeValues import MarkNodesWithShapeValues
 from mo.front.common.partial_infer.utils import int64_array, float32_array
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph, result, regular_op_with_empty_data, shaped_const_with_data, connect, \
-from mo.utils.unittest.graph import result, regular_op_with_empty_data, \
+    regular_op
-    shaped_const_with_data, connect, regular_op
 
 
 class TestMarkDataTypeInShapeOfSubgraphs(unittest.TestCase):
@@ -24,7 +23,7 @@ class TestMarkDataTypeInShapeOfSubgraphs(unittest.TestCase):
             **shaped_const_with_data('input', int64_array(inp_shape)),
             **regular_op_with_empty_data('shape', {'type': 'ShapeOf'}),
             **regular_op_with_empty_data('cast_to_float', {'type': 'Cast', 'dst_type': dst_type}),
-            **regular_op('mul_const',  {'op': 'Const'}),
+            **regular_op('mul_const', {'op': 'Const'}),
             **{'mul_const_d': {'kind': 'data', 'value': float32_array([1., 1., 1., 100.])}},
             **regular_op_with_empty_data('mul', {'type': 'Mul'}),
             **regular_op_with_empty_data('cast_to_int', {'type': 'Cast', 'dst_type': np.int64}),
@@ -38,7 +37,7 @@ class TestMarkDataTypeInShapeOfSubgraphs(unittest.TestCase):
             **regular_op_with_empty_data('cast_to_float', {'type': 'Cast', 'dst_type': dst_type,
                                                            'returns_shape_value': True}),
             **regular_op_with_empty_data('mul', {'type': 'Mul', 'returns_shape_value': True}),
-            **regular_op('mul_const',  {'op': 'Const', 'returns_shape_value': True}),
+            **regular_op('mul_const', {'op': 'Const', 'returns_shape_value': True}),
             **{'mul_const_d': {'kind': 'data', 'value': float32_array([1., 1., 1., 100.]),
                                'correct_data_type': True}},
             **regular_op_with_empty_data('cast_to_int', {'type': 'Cast', 'dst_type': np.int64,
@@ -69,21 +68,20 @@ class TestMarkDataTypeInShapeOfSubgraphs(unittest.TestCase):
 
     def test_run_with_const_input(self):
         inp_shape = (1, 3, 1000, 1000)
-        dst_type = np.float32
 
         nodes = {
             **shaped_const_with_data('input', int64_array(inp_shape)),
-            **regular_op('sizes_const',  {'op': 'Const'}),
+            **regular_op('sizes_const', {'op': 'Const'}),
             **{'sizes_const_d': {'kind': 'data', 'value': float32_array([1., 1., 1., 100.])}},
-             **regular_op_with_empty_data('interpolate', {'type': 'Interpolate', 'shape_calculation_model': 'scales'}),
+            **regular_op_with_empty_data('interpolate', {'type': 'Interpolate', 'shape_calculation_model': 'scales'}),
             **result('res'),
         }
 
         nodes_ref = {
             **shaped_const_with_data('input', int64_array(inp_shape)),
-            **regular_op('sizes_const',  {'op': 'Const', 'returns_shape_value': True}),
+            **regular_op('sizes_const', {'op': 'Const', 'returns_shape_value': True}),
             **{'sizes_const_d': {'kind': 'data', 'value': float32_array([1., 1., 1., 100.])}},
-              **regular_op_with_empty_data('interpolate', {'type': 'Interpolate', 'shape_calculation_model': 'scales'}),
+            **regular_op_with_empty_data('interpolate', {'type': 'Interpolate', 'shape_calculation_model': 'scales'}),
             **result('res'),
         }
 

model-optimizer/unit_tests/extensions/back/MatMulNormalizer_test.py

@@ -11,9 +11,9 @@ from extensions.ops.MatMul import MatMul
 from mo.front.common.partial_infer.utils import int64_array
 from mo.ops.reshape import Reshape
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, \
+from unit_tests.utils.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, \
     result, connect
-from mo.utils.unittest.graph import regular_op_with_empty_data as op_with_empty_data
+from unit_tests.utils.graph import regular_op_with_empty_data as op_with_empty_data
 
 
 @generator

model-optimizer/unit_tests/extensions/back/ResultRename_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.back.ResultRename import ResultRename
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op, result
+from unit_tests.utils.graph import build_graph, regular_op, result
 
 nodes = {
     **regular_op('Op1', {'type': 'Op1', 'kind': 'op', 'op': 'Op1'}),

model-optimizer/unit_tests/extensions/back/ShuffleChannelPatternOptimization_test.py

@@ -14,7 +14,7 @@ from extensions.ops.transpose import Transpose
 from mo.front.common.partial_infer.utils import int64_array
 from mo.ops.reshape import Reshape
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, \
     valued_const_with_data, connect, regular_op_with_empty_data
 
 

model-optimizer/unit_tests/extensions/back/ShufflenetReLUReorder_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.back.ShufflenetReLUReorder import ShufflenetReLUReorder
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 # The dictionary with nodes attributes used to build various graphs. A key is the name of the node and the value is the
 # dictionary with node attributes.

model-optimizer/unit_tests/extensions/back/SpecialNodesFinalization_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.back.SpecialNodesFinalization import CreateConstNodesReplacement, RemoveConstToResult
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph_with_attrs
+from unit_tests.utils.graph import build_graph_with_attrs
 
 
 class CreateConstNodesReplacementTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/TransposeReduceFusing_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.back.TransposeReduceFusing import TransposeReduce
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     # op

model-optimizer/unit_tests/extensions/back/compress_quantized_weights_test.py

@@ -13,7 +13,7 @@ from extensions.ops.elementwise import Sub, Mul
 from extensions.ops.fakequantize import FakeQuantize
 from mo.front.common.partial_infer.eltwise import eltwise_infer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, result, connect, \
+from unit_tests.utils.graph import build_graph, regular_op_with_shaped_data, valued_const_with_data, result, connect, \
     shaped_const_with_data
 
 

model-optimizer/unit_tests/extensions/back/insert_compatibility_l2normalization_test.py

@@ -6,7 +6,7 @@ import unittest
 import numpy as np
 
 from extensions.back.insert_compatibility_l2normalization import CompatibilityL2NormalizationPattern
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class CompatibilityL2NormalizationPatternTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/kaldi_remove_memory_output_test.py

@@ -4,7 +4,7 @@
 import unittest
 
 from extensions.back.kaldi_remove_memory_output import KaldiRemoveMemoryOutputBackReplacementPattern
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class KaldiRemoveMemoryOutputTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/back/remove_last_softmax_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.back.remove_last_softmax_pattern import RemoveLastSoftMaxPattern, RemoveLastLogSoftMaxPattern
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class KaldiRemoveLastSoftMaxTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/ATenToEmbeddingBag_test.py

@@ -8,8 +8,7 @@ import numpy as np
 from extensions.front.ATenToEmbeddingBag import AtenToEmbeddingBag
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, \
+from unit_tests.utils.graph import build_graph, result, regular_op, const
-    regular_op, const
 
 
 class AtenToEmbeddingBagTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/AttributedClampNormalizer_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.AttributedClampNormalizer import AttributedClampNormalizer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const
+from unit_tests.utils.graph import build_graph, const
 
 nodes_attributes = {
     'placeholder': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/AttributedPadToPad_test.py

@@ -8,7 +8,7 @@ import numpy as np
 from extensions.front.AttributedPadToPad import AttributedPadToPad
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const
+from unit_tests.utils.graph import build_graph, const
 
 nodes_attributes = {
     'placeholder': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/AttributedRollToRoll_test.py

@@ -7,7 +7,7 @@ from extensions.front.AttributedRollToRoll import AttributedRollToRoll
 from mo.front.common.partial_infer.utils import int64_array
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, result, regular_op
+from unit_tests.utils.graph import build_graph, const, result, regular_op
 
 nodes_attributes = {
     **regular_op('placeholder', {'type': 'Parameter'}),

model-optimizer/unit_tests/extensions/front/DropoutWithRandomUniformReplacer_test.py

@@ -7,7 +7,7 @@ import unittest
 
 from extensions.front.DropoutWithRandomUniformReplacer import DropoutWithRandomUniformReplacer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op
+from unit_tests.utils.graph import build_graph, result, regular_op
 
 
 class DropoutWithRandomUniformReplacerTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/GeLUMerger_Erf_test.py

@@ -7,7 +7,7 @@ from math import sqrt
 from extensions.front.GeLUMerger_Erf import GeLUMergerErf
 from mo.front.common.partial_infer.utils import float_array, int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, regular_op, result, build_graph
+from unit_tests.utils.graph import const, regular_op, result, build_graph
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('gelu', {'type': 'Gelu', 'approximation_mode': 'erf', 'name': 'final_mul'}),

model-optimizer/unit_tests/extensions/front/GeLUMerger_Tanh_test.py

@@ -8,7 +8,7 @@ import numpy as np
 
 from extensions.front.GeLUMerger_Tanh import GeLUMergerTanh
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes_erf = {
     'inp': {'kind': 'op', 'op': 'AnyOp'},

model-optimizer/unit_tests/extensions/front/HSigmoid_fusion_test.py

@@ -7,7 +7,7 @@ from extensions.front.HSigmoid_fusion import HSigmoidWithClamp, HSigmoidWithMinM
     HSigmoidWithReluMul
 from mo.front.common.partial_infer.utils import float_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('hsigmoid', {'type': 'HSigmoid', 'name': 'final_mul'}),

model-optimizer/unit_tests/extensions/front/HSwish_fusing_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.front.HSwish_fusion import HSwishWithClamp, HSwishWithMinMax
 from mo.front.common.partial_infer.utils import float_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('hswish', {'type': 'HSwish', 'name': 'final_mul'}),

model-optimizer/unit_tests/extensions/front/LayerNorm_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.LayerNorm import LayerNorm
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class TestMVNPatternReplacement(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/Log1p_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.Log1p import Log1p
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'placeholder': {'shape': np.array([4, 5, 6]), 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/Mish_fusion_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.Mish_fusion import MishFusion
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op, result, build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph, regular_op, result, build_graph_with_edge_attrs
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('mish', {'type': 'Mish', 'name': 'final_mul'}),

model-optimizer/unit_tests/extensions/front/OneHotDepthNormalizer_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.front.OneHotDepthNormalizer import OneHotDepthNormalizer
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, \
+from unit_tests.utils.graph import build_graph, result, \
     regular_op, const
 
 

model-optimizer/unit_tests/extensions/front/Pack_test.py

@@ -9,7 +9,7 @@ from generator import generator, generate
 from extensions.front.Pack import Pack
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'placeholder_0': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/RollWithEmptyAxesReplacer_test.py

@@ -7,7 +7,7 @@ from extensions.front.RollWithEmptyAxesReplacer import RollWithEmptyAxesReplacer
 from mo.front.common.partial_infer.utils import int64_array
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, result, regular_op
+from unit_tests.utils.graph import build_graph, const, result, regular_op
 
 nodes_attributes = {
     **regular_op('placeholder', {'type': 'Parameter'}),

model-optimizer/unit_tests/extensions/front/Softplus_fusion_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.front.Softplus_fusion import SoftplusFusion
 from mo.front.common.partial_infer.utils import float_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph, const, regular_op, result, build_graph_with_edge_attrs
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('softplus', {'type': 'SoftPlus', 'name': 'final_log'}),

model-optimizer/unit_tests/extensions/front/Swish_fusion_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.Swish_fusion import SwishWithSigmoidWithoutBeta, SwishWithSigmoidWithBeta
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op, result, build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph, regular_op, result, build_graph_with_edge_attrs
 
 ref_nodes = {**regular_op('input', {'type': 'Parameter'}),
              **regular_op('swish', {'type': 'Swish', 'name': 'final_mul'}),

model-optimizer/unit_tests/extensions/front/ThresholdedReluDecomposition_test.py

@@ -6,7 +6,7 @@ import unittest
 from extensions.front.ThresholdedReluDecomposition import ThresholdedReluDecomposition
 from mo.front.common.partial_infer.utils import float_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, const
+from unit_tests.utils.graph import build_graph, const
 
 nodes_attributes = {
     'parameter': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/binary_quantize_normalization_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.binary_quantize_normalization import BinaryFakeQuantizeNormalization
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 graph_nodes = {
     '0': {'name': 'input', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/broadcast_with_range_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.broadcast_with_range import ExpandRangeConstant
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
     regular_op_with_empty_data, connect_data
 
 

model-optimizer/unit_tests/extensions/front/caffe/MVNCaffeToMVN_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.caffe.MVNCaffeToMVN import MVNCaffeToMVN
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op_with_empty_data, result, const, connect_front
+from unit_tests.utils.graph import build_graph, regular_op_with_empty_data, result, const, connect_front
 
 nodes = {
     **regular_op_with_empty_data('input', {'type': 'Parameter'}),

model-optimizer/unit_tests/extensions/front/caffe/accum_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.accum_ext import AccumFrontExtractor
 from extensions.ops.accum import AccumOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeAccumProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/argmax_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.argmax_ext import ArgMaxFrontExtractor
 from extensions.ops.argmax import ArgMaxOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeArgMaxProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/axpy_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.caffe.axpy import AxpyToSSandAdd
 from mo.graph.graph import Node
-from mo.utils.unittest.graph import build_graph_with_edge_attrs
+from unit_tests.utils.graph import build_graph_with_edge_attrs
 
 
 class TestAxpyReplacer(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/caffe/bn_test.py

@@ -7,8 +7,8 @@ import unittest
 from extensions.front.caffe.bn import BNToScaleShift
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.extractors import FakeParam
+from unit_tests.utils.extractors import FakeParam
-from mo.utils.unittest.graph import build_graph_with_edge_attrs, build_graph_with_attrs
+from unit_tests.utils.graph import build_graph_with_edge_attrs, build_graph_with_attrs
 
 
 class FakeBNProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/conv_ext_test.py

@@ -9,7 +9,7 @@ import numpy as np
 from extensions.front.caffe.conv_ext import ConvFrontExtractor, DeconvFrontExtractor, conv_create_attrs, conv_set_params
 from mo.front.caffe.extractors.utils import get_list_from_container
 from mo.utils.error import Error
-from mo.utils.unittest.extractors import PB, FakeParam, FakeMultiParam
+from unit_tests.utils.extractors import PB, FakeParam, FakeMultiParam
 
 
 class FakeConvProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/correlation_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.correlation_ext import CorrelationFrontExtractor
 from extensions.ops.correlation import CorrelationOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeCorrProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/crop_ext_test.py

@@ -8,8 +8,8 @@ from extensions.front.caffe.crop_ext import CropFrontExtractor
 from mo.front.common.partial_infer.crop import crop_infer
 from mo.ops.crop import Crop
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeCropProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/ctcgreedydecoder_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.ctcgreedydecoder_ext import CTCGreedyDecoderFrontExtractor
 from extensions.ops.ctc_greedy_decoder import CTCGreedyDecoderOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeCTCGreedyDecoderProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/data_augmentation_ext_test.py

@@ -9,8 +9,8 @@ import numpy as np
 from extensions.front.caffe.data_augmentation_ext import DataAugmentationFrontExtractor
 from extensions.ops.data_augmentation import DataAugmentationOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeDAProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/elementwise_ext_test.py

@@ -5,8 +5,8 @@ import unittest
 from unittest.mock import patch
 
 from extensions.front.caffe.elementwise_ext import BiasToAdd
-from mo.utils.unittest.extractors import FakeModelLayer, FakeMultiParam
+from unit_tests.utils.extractors import FakeModelLayer, FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeBiasProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/eltwise_add_normalize_test.py

@@ -7,8 +7,8 @@ import numpy as np
 
 from extensions.front.caffe.eltwise_add_normalize import EltwiseAddNormalize
 from mo.front.common.partial_infer.utils import int64_array
-from mo.middle.passes.eliminate_test import build_graph
 from mo.utils.ir_engine.compare_graphs import compare_graphs
+from unit_tests.utils.graph import build_graph
 
 input_shape = int64_array([1, 4, 10])
 const_1_value = np.array([2.0])

model-optimizer/unit_tests/extensions/front/caffe/elu_test.py

@@ -5,8 +5,8 @@ import unittest
 from unittest.mock import patch
 
 from extensions.front.caffe.elu import ELUFrontExtractor
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/grn_ext_test.py

@@ -8,8 +8,8 @@ from extensions.front.caffe.grn_ext import GRNFrontExtractor
 from extensions.ops.grn import GRNOp
 from mo.front.common.partial_infer.elemental import copy_shape_infer
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeGRNProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/normalize_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.normalize_ext import NormalizeFrontExtractor
 from extensions.ops.normalize import NormalizeOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeNormalizeProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/pooling_ext_test.py

@@ -8,7 +8,7 @@ import numpy as np
 from extensions.front.caffe.pooling_ext import PoolingFrontExtractor
 from mo.front.common.extractors.utils import layout_attrs
 from mo.ops.pooling import Pooling
-from mo.utils.unittest.extractors import PB, FakeMultiParam
+from unit_tests.utils.extractors import PB, FakeMultiParam
 
 
 class FakeProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/power_file_ext_test.py

@@ -8,8 +8,8 @@ from extensions.front.caffe.power_file_ext import PowerFileFrontExtractor
 from extensions.ops.power_file import PowerFileOp
 from mo.front.common.partial_infer.elemental import copy_shape_infer
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakePowerFileProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/prelu_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.prelu_ext import PreluFrontExtractor
 from extensions.ops.prelu import PReLU
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakePReLUProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/priorbox_clustered_ext_test.py

@@ -9,8 +9,8 @@ import numpy as np
 from extensions.front.caffe.priorbox_clustered_ext import PriorBoxClusteredFrontExtractor
 from extensions.ops.priorbox_clustered import PriorBoxClusteredOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakePriorBoxClusteredProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/priorbox_ext_test.py

@@ -9,8 +9,8 @@ import numpy as np
 from extensions.front.caffe.priorbox_ext import PriorBoxFrontExtractor
 from extensions.ops.priorbox import PriorBoxOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam, FakeParam
+from unit_tests.utils.extractors import FakeMultiParam, FakeParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeMultiParamListFields(FakeMultiParam):

model-optimizer/unit_tests/extensions/front/caffe/proposal_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.proposal_ext import ProposalFrontExtractor
 from extensions.ops.proposal import ProposalOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode, FakeAttr
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeProposalProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/proposal_python_ext_test.py

@@ -6,8 +6,8 @@ import unittest
 from extensions.front.caffe.proposal_python_ext import ProposalPythonFrontExtractor
 from extensions.ops.proposal import ProposalOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode, FakeAttr
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeProposalPythonProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/regionyolo_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.regionyolo_ext import RegionYoloFrontExtractor
 from extensions.ops.regionyolo import RegionYoloOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeRegionYoloProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/reorgyolo_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.reorgyolo_ext import ReorgYoloFrontExtractor
 from extensions.ops.reorgyolo import ReorgYoloOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeReorgYoloProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/simplernms_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.simplernms_ext import SimplerNMSFrontExtractor
 from extensions.ops.simplernms import SimplerNMSOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeSimplerNMSProtoLayer:

model-optimizer/unit_tests/extensions/front/caffe/spatial_transformer_ext_test.py

@@ -7,8 +7,8 @@ from unittest.mock import patch
 from extensions.front.caffe.spatial_transformer_ext import SpatialTransformFrontExtractor
 from extensions.ops.spatial_transformer import SpatialTransformOp
 from mo.ops.op import Op
-from mo.utils.unittest.extractors import FakeMultiParam
+from unit_tests.utils.extractors import FakeMultiParam
-from mo.utils.unittest.graph import FakeNode
+from unit_tests.utils.graph import FakeNode
 
 
 class FakeSpatialTransformProtoLayer:

model-optimizer/unit_tests/extensions/front/div_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.div import Div
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
     connect_data
 
 nodes = {

model-optimizer/unit_tests/extensions/front/eltwise_n_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.eltwise_n import EltwiseNReplacement
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'placeholder_1': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/freeze_placeholder_value_test.py

@@ -6,7 +6,7 @@ import unittest
 import numpy as np
 
 from extensions.front.freeze_placeholder_value import FreezePlaceholderValue
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_bool = {
     '0': {'name': 'input1', 'kind': 'op', 'op': 'Parameter', 'data_type': bool, 'shape': np.array([])},

model-optimizer/unit_tests/extensions/front/image_scaler_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.image_scaler import ImageScaler
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'placeholder_1': {'shape': None, 'type': 'Parameter', 'kind': 'op', 'op': 'Parameter'},

model-optimizer/unit_tests/extensions/front/instance_normalization_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.instance_normalization import InstanceNormalization
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 nodes_attributes = {
     'input': {'kind': 'op', 'op': 'AnyOp'},

model-optimizer/unit_tests/extensions/front/interpolate_reshape_test.py

@@ -9,7 +9,7 @@ from generator import generator, generate
 from extensions.front.interpolate_reshape import InterpolateWithConcat
 from mo.front.common.partial_infer.utils import int64_array
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
+from unit_tests.utils.graph import build_graph, result, regular_op_with_shaped_data, valued_const_with_data, connect, \
     connect_data
 
 nodes = {

model-optimizer/unit_tests/extensions/front/kaldi/add_permute_after_convolution_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.kaldi.add_permute_after_convolution import ReplaceConvolutionTranspose
 from mo.graph.graph import Node
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class ReplaceConvolutionTransposeTests(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/kaldi/apply_counts_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.kaldi.apply_counts import apply_biases_to_last_layer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class TestKaldiPipeline(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/kaldi/memory_offset_adjustment_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.kaldi.memory_offset_adjustment import MemoryOffsetAdjustment
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class MemoruOffsetAdjustmentTests(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/kaldi/replace_lstm_nonlinearity_test.py

@@ -8,7 +8,7 @@ import numpy as np
 from extensions.front.kaldi.replace_lstm_nonlinearity import ReplaceLstmNonLinearityPattern
 from mo.graph.graph import Node
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class ReplaceLstmNonlinearityTests(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/kaldi/sigmoid_ext_test.py

@@ -3,7 +3,7 @@
 
 from extensions.front.kaldi.sigmoid_ext import SigmoidFrontExtractor
 from extensions.ops.activation_ops import Sigmoid
-from mo.front.kaldi.extractors.common_ext_test import KaldiFrontExtractorTest
+from unit_tests.mo.front.kaldi.extractors.common_ext_test import KaldiFrontExtractorTest
 from mo.ops.op import Op
 
 

model-optimizer/unit_tests/extensions/front/kaldi/tanh_ext_test.py

@@ -3,7 +3,7 @@
 
 from extensions.front.kaldi.tanh_component_ext import TanhFrontExtractor
 from extensions.ops.activation_ops import Tanh
-from mo.front.kaldi.extractors.common_ext_test import KaldiFrontExtractorTest
+from unit_tests.mo.front.kaldi.extractors.common_ext_test import KaldiFrontExtractorTest
 from mo.ops.op import Op
 
 

model-optimizer/unit_tests/extensions/front/kaldi/tdnn_component_replacer_test.py

@@ -8,7 +8,7 @@ from generator import generator, generate
 
 from extensions.front.kaldi.tdnn_component_replacer import TdnnComponentReplacer
 from mo.utils.ir_engine.compare_graphs import compare_graphs
-from mo.utils.unittest.graph import build_graph, regular_op, result, connect_front, const
+from unit_tests.utils.graph import build_graph, regular_op, result, connect_front, const
 
 
 @generator

model-optimizer/unit_tests/extensions/front/mxnet/RNN_ext_test.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from extensions.front.mxnet.RNN_ext import RNNFrontExtractor
 from mo.utils.error import Error
-from mo.utils.unittest.extractors import PB
+from unit_tests.utils.extractors import PB
 
 
 class RNNFrontExtractorTest(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/mxnet/activation_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.mxnet.activation import ActivationFrontExtractor
 from mo.graph.graph import Node
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class TestActivationFrontExtractorOp(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/mxnet/add_input_data_to_prior_boxes_test.py

@@ -8,7 +8,7 @@ import numpy as np
 
 from extensions.front.mxnet.add_input_data_to_prior_boxes import AddInputDataToPriorBoxes
 from mo.graph.graph import Node
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class TestMxnetPipeline(unittest.TestCase):

model-optimizer/unit_tests/extensions/front/mxnet/check_softmax_node_inputs_test.py

@@ -5,7 +5,7 @@ import unittest
 
 from extensions.front.mxnet.check_softmax_node_inputs import CheckSoftmaxNodeInputs
 from mo.graph.graph import Node
-from mo.utils.unittest.graph import build_graph
+from unit_tests.utils.graph import build_graph
 
 
 class TestCheckSoftmaxNodeInputs(unittest.TestCase):