|
|
|
|
|
import pytest |
|
|
|
|
|
from mmocr.utils import StringStrip |
|
|
|
|
|
|
|
|
def test_string_strip(): |
|
|
strip_list = [True, False] |
|
|
strip_pos_list = ['both', 'left', 'right'] |
|
|
strip_str_list = [None, ' '] |
|
|
|
|
|
in_str_list = [ |
|
|
' hello ', 'hello ', ' hello', ' hello', 'hello ', 'hello ', 'hello', |
|
|
'hello', 'hello', 'hello', 'hello', 'hello' |
|
|
] |
|
|
out_str_list = [ |
|
|
'hello', 'hello', 'hello', 'hello', 'hello', 'hello', 'hello', 'hello', |
|
|
'hello', 'hello', 'hello', 'hello' |
|
|
] |
|
|
|
|
|
for idx1, strip in enumerate(strip_list): |
|
|
for idx2, strip_pos in enumerate(strip_pos_list): |
|
|
for idx3, strip_str in enumerate(strip_str_list): |
|
|
tmp_args = dict( |
|
|
strip=strip, strip_pos=strip_pos, strip_str=strip_str) |
|
|
strip_class = StringStrip(**tmp_args) |
|
|
i = idx1 * len(strip_pos_list) * len( |
|
|
strip_str_list) + idx2 * len(strip_str_list) + idx3 |
|
|
|
|
|
assert strip_class(in_str_list[i]) == out_str_list[i] |
|
|
|
|
|
with pytest.raises(AssertionError): |
|
|
StringStrip(strip='strip') |
|
|
StringStrip(strip_pos='head') |
|
|
StringStrip(strip_str=['\n', '\t']) |
|
|
|
