From 9a58fecc197e7ea18c61a2b704779013c6c5c2ae Mon Sep 17 00:00:00 2001 From: Arseny Date: Tue, 30 Jul 2024 21:03:28 +0200 Subject: [PATCH] test: squeeze-21 --- tests/node_converters/squeeze_test.py | 35 ++++++++++----------------- 1 file changed, 13 insertions(+), 22 deletions(-) diff --git a/tests/node_converters/squeeze_test.py b/tests/node_converters/squeeze_test.py index dd04cfee..e96e1597 100644 --- a/tests/node_converters/squeeze_test.py +++ b/tests/node_converters/squeeze_test.py @@ -55,30 +55,21 @@ def _test_squeeze( @pytest.mark.filterwarnings('ignore::torch.jit._trace.TracerWarning') +@pytest.mark.parametrize('opset_version', [11, 13, 21]) @pytest.mark.parametrize( - 'shape,axes,opset_version', + 'shape, axes', ( - ([1, 3, 4, 5], [0], 11), - ([1, 3, 1, 5], [-2], 11), - ([1, 3, 1, 5], [0, 2], 11), - ([1, 3, 1, 5], [2, 0], 11), - ([1, 3, 1, 1, 1, 5, 1], [2, 0, 6], 11), - ([1, 3, 1, 5], [0, -2], 11), - ([1, 3, 1, 5], [-2, 0], 11), - ([1, 3, 1, 5], None, 11), - ([1, 1, 1, 1], None, 11), - ([1], None, 11), - ([3, 3, 3], None, 11), - ([1, 3, 4, 5], [0], 13), - ([1, 3, 1, 5], [-2], 13), - ([1, 3, 1, 5], [0, 2], 13), - ([1, 3, 1, 5], [2, 0], 13), - ([1, 3, 1, 5], [0, -2], 13), - ([1, 3, 1, 5], [-2, 0], 13), - ([1, 3, 1, 5], None, 13), - ([1, 1, 1, 1], None, 13), - ([1], None, 13), - ([3, 3, 3], None, 13), + ([1, 3, 4, 5], [0]), + ([1, 3, 1, 5], [-2]), + ([1, 3, 1, 5], [0, 2]), + ([1, 3, 1, 5], [2, 0]), + ([1, 3, 1, 1, 1, 5, 1], [2, 0, 6]), + ([1, 3, 1, 5], [0, -2]), + ([1, 3, 1, 5], [-2, 0]), + ([1, 3, 1, 5], None), + ([1, 1, 1, 1], None), + ([1], None), + ([3, 3, 3], None), ), ) def test_squeeze( # pylint: disable=missing-function-docstring