15 lines
335 B
Python
15 lines
335 B
Python
import pytest
|
|
|
|
from atst.forms.financial import suggest_pe_id
|
|
|
|
|
|
@pytest.mark.parametrize("input_,expected", [
|
|
('0603502N', None),
|
|
('0603502NZ', None),
|
|
('603502N', '0603502N'),
|
|
('063502N', '0603502N'),
|
|
('63502N', '0603502N'),
|
|
])
|
|
def test_suggest_pe_id(input_, expected):
|
|
assert suggest_pe_id(input_) == expected
|