diff --git a/tests/forms/test_financial.py b/tests/forms/test_financial.py index 925aec2a..b1f08b64 100644 --- a/tests/forms/test_financial.py +++ b/tests/forms/test_financial.py @@ -2,21 +2,21 @@ import pytest from werkzeug.datastructures import ImmutableMultiDict from atst.forms.financial import FinancialForm, ExtendedFinancialForm -from atst.eda_client import MockEDAClient +from atst.domain.requests.financial_verification import PENumberValidator -# @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 +@pytest.mark.parametrize( + "input_,expected", + [ + ("0603502N", None), + ("0603502NZ", None), + ("603502N", "0603502N"), + ("063502N", "0603502N"), + ("63502N", "0603502N"), + ], +) +def test_suggest_pe_id(input_, expected): + assert PENumberValidator().suggest_pe_id(input_) == expected def test_funding_type_other_not_required_if_funding_type_is_not_other():