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

generator to pytest (#19298) 

* trying with pytest

* update as per suggested

---------

Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>
This commit is contained in:
Pratham Ingawale 2023-08-22 19:12:56 +05:30 committed by GitHub
parent 6cd70af7c2
commit 9a76daf94b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tools/mo/unit_tests/mo/back/ChangeRandomUniformOutputType_test.py
View File

@ -1,11 +1,9 @@
# Copyright (C) 2018-2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
import unittest
from argparse import Namespace
import pytest
import numpy as np
from generator import generator, generate
from openvino.tools.mo.back.ChangeRandomUniformOutputType import ChangeRandomUniformOutputType
from openvino.tools.mo.graph.graph import Node
@ -31,15 +29,14 @@ edges_with_convert = [*connect('placeholder', '0:random_uniform'), *connect('min
*connect('convert', 'result'), ]
@generator
class ChangeRandomUniformOutputTypeTest(unittest.TestCase):
@generate(*[
("FP16", np.float32, np.float16),
("FP32", np.float16, np.float32),
("FP32", np.float32, None),
("FP32", np.int64, None)
])
def test_change_random_uniform_output_type(self, ir_type, out_type, dst_type):
class TestChangeRandomUniformOutputType():
@pytest.mark.parametrize("ir_type, out_type, dst_type", [
("FP16", np.float32, np.float16),
("FP32", np.float16, np.float32),
("FP32", np.float32, None),
("FP32", np.int64, None)
])
def test_change_random_uniform_output_type(self,ir_type, out_type, dst_type):
graph = build_graph(nodes, edges, cli=Namespace(data_type=ir_type))
graph_ref = build_graph(nodes, edges if dst_type is None else edges_with_convert, {},
nodes_with_edges_only=True)
@ -48,8 +45,8 @@ class ChangeRandomUniformOutputTypeTest(unittest.TestCase):
ChangeRandomUniformOutputType().find_and_replace_pattern(graph)
(flag, resp) = compare_graphs(graph, graph_ref, 'result', check_op_attrs=True)
self.assertTrue(flag, resp)
assert flag, resp
if dst_type is not None:
convert_node = Node(graph, 'random_uniform').out_port(0).get_destination().node
self.assertTrue(convert_node['dst_type'] == dst_type)
assert convert_node['dst_type'] == dst_type