pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify BA code valdiation

Browse Source
This commit is contained in:
richard-dds 2018-08-24 12:01:34 -04:00
parent 52f577aea6
commit 27601300aa
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
atst/forms/financial.py
View File

@ -24,7 +24,7 @@ PE_REGEX = re.compile(
TREASURY_CODE_REGEX = re.compile(r"^0*([1-9]{4}|[1-9]{6})$") TREASURY_CODE_REGEX = re.compile(r"^0*([1-9]{4}|[1-9]{6})$")
BA_CODE_REGEX = re.compile(r"^0*[1-9]{2}\w?$") BA_CODE_REGEX = re.compile(r"[0-9]{2}\w?$")
def suggest_pe_id(pe_id): def suggest_pe_id(pe_id):

16
tests/forms/test_financial.py
View File

@ -57,13 +57,19 @@ def test_treasury_code_validation(input_, expected):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"input_,expected", "input_,expected",
[ [
("1", False),
("12", True), ("12", True),
("00012", True), ("01", True),
("0A", False),
("A", False),
("AB", False),
("123", True),
("012", True),
("12A", True), ("12A", True),
("000123", True), ("02A", True),
("00012A", True), ("0012", False),
("0001", False), ("012A", False),
("00012AB", False), ("2AB", False)
], ],
) )
def test_ba_code_validation(input_, expected): def test_ba_code_validation(input_, expected):