From f56dd5621eb4912c8b5ffe0964286f1bcee87fef Mon Sep 17 00:00:00 2001 From: richard-dds Date: Tue, 16 Oct 2018 14:15:27 -0400 Subject: [PATCH] Revive suggest_pe_id tests --- tests/forms/test_financial.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) 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():