diff --git a/ciprs_reader/parser/lines.py b/ciprs_reader/parser/lines.py index 76592be..4b4c0be 100644 --- a/ciprs_reader/parser/lines.py +++ b/ciprs_reader/parser/lines.py @@ -39,11 +39,10 @@ def clean(self, matches): data = {} fileno = self.report["General"].get("File No", "") if fileno: - if "CR" in fileno: - if "CRS" in fileno: - data["Superior"] = "Yes" - else: - data["District"] = "Yes" + if "CRS" in fileno: + data["Superior"] = "Yes" + elif "CR" in fileno or "IF" in fileno: + data["District"] = "Yes" return data def extract(self, matches, report): diff --git a/tests/test_parsers.py b/tests/test_parsers.py index 963ef1f..1a0b079 100644 --- a/tests/test_parsers.py +++ b/tests/test_parsers.py @@ -11,7 +11,7 @@ def test_defendent_dob(report, state): def test_court_type_other(report, state): - report = {"General": {"File No": "11IF777777"}} + report = {"General": {"File No": "11XX777777"}} matches = parsers.DistrictSuperiorCourt(report, state).match("") assert matches is not None, "Regex match failed" assert matches == {} @@ -26,6 +26,15 @@ def test_court_type_cr(report, state): assert report["General"]["District"] == "Yes" +def test_court_type_if(report, state): + report = {"General": {"File No": "11IF777777"}} + parser = parsers.DistrictSuperiorCourt(report, state) + parser.find("") + assert parser.matches is not None, "Regex match failed" + assert parser.matches == {"District": "Yes"} + assert report["General"]["District"] == "Yes" + + def test_court_type_crs(report, state): report = {"General": {"File No": "11CRS777777"}} parser = parsers.DistrictSuperiorCourt(report, state)