Start parsing offense record count

deardurham · Apr 30, 2023 · d278019 · d278019
1 parent cbf6903
commit d278019
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -66,7 +66,7 @@ pip install -e .
 Read CIPRS PDF:
 
 ```
-python ciprs_-eader.py ./cypress-example.pdf
+python ciprs_reader.py ./cypress-example.pdf
 ```
 
 Run Jupyter:

diff --git a/ciprs_reader/parser/section/offense.py b/ciprs_reader/parser/section/offense.py
@@ -50,6 +50,7 @@ def clean(self, matches):
 
     def extract(self, matches, report):
         record = {
+            "Count": matches["num"],
             "Action": matches["action"],
             "Description": matches["desc"],
             "Severity": matches["severity"],

diff --git a/tests/parsers/test_offense.py b/tests/parsers/test_offense.py
@@ -21,6 +21,7 @@ def test_offense_record_charged_with_number(report, state):
     """
     matches = offense.OffenseRecordRowWithNumber(report, state).match(string)
     assert matches is not None, "Regex match failed"
+    assert matches["count"] == "54"
     assert matches["action"] == "CHARGED"
     assert matches["desc"] == "SPEEDING(80 mph in a 65 mph zone)"
     assert matches["severity"] == "INFRACTION"