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

[TF FE] Extend StringSplitV2 layer tests and fix warnings (#24148) 

**Details:** Extend StringSplitV2 layer tests and fix warnings. Merge
after https://github.com/openvinotoolkit/openvino_tokenizers/pull/121

**Ticket:** TBD

---------

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2024-04-22 01:23:24 +04:00 committed by GitHub
parent 254296e4a0
commit 5a18bc1277
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/layer_tests/tensorflow_tests/test_tf_Equal.py
View File

@ -223,7 +223,7 @@ class TestEqualStr(CommonTFLayerTest):
x_shape = inputs_info['x:0']
y_shape = inputs_info['y:0']
inputs_data = {}
strings_dictionary = ['UPPER<>CASE SENTENCE', 'lower case\n\s sentence', ' UppEr LoweR CAse SENtence \t\n',
strings_dictionary = ['UPPER<>CASE SENTENCE', 'lower case\n sentence', ' UppEr LoweR CAse SENtence \t\n',
' some sentence', 'another sentence HERE ']
inputs_data['x:0'] = rng.choice(strings_dictionary, x_shape)
inputs_data['y:0'] = rng.choice(strings_dictionary, y_shape)

2
tests/layer_tests/tensorflow_tests/test_tf_StaticRegexReplace.py
View File

@ -36,7 +36,7 @@ class TestStaticRegexReplace(CommonTFLayerTest):
return tf_net, ref_net
@pytest.mark.parametrize('input_shape', [[], [2], [3, 4], [1, 3, 2]])
@pytest.mark.parametrize('pattern', ['(\s)|(-)', '[A-Z]{2,}', '^\s+|\s+$'])
@pytest.mark.parametrize('pattern', [r'(\s)|(-)', r'[A-Z]{2,}', r'^\s+|\s+$'])
@pytest.mark.parametrize('rewrite', ['', 'replacement word'])
@pytest.mark.parametrize('replace_global', [None, True, False])
@pytest.mark.precommit

4
tests/layer_tests/tensorflow_tests/test_tf_StringSplitV2.py
View File

@ -17,7 +17,7 @@ class TestStringSplitV2(CommonTFLayerTest):
assert 'input:0' in inputs_info
input_shape = inputs_info['input:0']
inputs_data = {}
strings_dictionary = ['UPPER<>CASE SENTENCE', 'lower case\n\s sentence', ' UppEr LoweR CAse SENtence \t\n',
strings_dictionary = ['UPPER<>CASE SENTENCE', 'lower case\n sentence', ' UppEr LoweR CAse SENtence \t\n',
' some sentence', 'another sentence HERE ']
inputs_data['input:0'] = rng.choice(strings_dictionary, input_shape)
return inputs_data
@ -39,7 +39,7 @@ class TestStringSplitV2(CommonTFLayerTest):
@pytest.mark.parametrize('input_shape', [[1], [2], [5]])
@pytest.mark.parametrize('sep', ['', '<>'])
@pytest.mark.parametrize('maxsplit', [None, -1])
@pytest.mark.parametrize('maxsplit', [None, -1, 5, 10])
@pytest.mark.precommit
@pytest.mark.nightly
@pytest.mark.xfail(condition=platform.system() in ('Darwin', 'Linux') and platform.machine() in ['arm', 'armv7l',