add all agencies argument

Author: jnelson16

regcensus/api.py

@@ -24,7 +24,8 @@ def get_values(series, jurisdiction, date, filtered=True, summary=True,
         summary (optional): Return summary instead of document level data
         filtered (optional): Exclude poorly-performing industry results
         documentType (optional): ID for type of document
-        agency (optional): Agency ID
+        agency (optional): Agency ID (use 'all' for all agencies,
+            only works for a single jurisdiction)
         industry (optional): Industry code using the jurisdiction-specific
             coding system (use 'all' for all industries)
         dateIsRange (optional): Indicating whether the time parameter is range
@@ -65,6 +66,9 @@ def get_values(series, jurisdiction, date, filtered=True, summary=True,
         pp.pprint(list_jurisdictions())
         return
 
+    # Allows for all agency data to be returned
+    if str(agency).lower() == 'all':
+        agency = list(list_agencies(jurisdiction).values())
     # If multiple agencies are given, parses the list into a string
     if type(agency) == list:
         url_call += f'&agency={",".join(str(i) for i in agency)}'

tests/test_api.py

@@ -150,6 +150,15 @@ def test_get_values_agency():
     assert order_results(results, 'seriesValue') == [62.0]
 
 
+def test_get_values_all_agencies():
+    results = rc.get_values(
+        series=91, jurisdiction=38, date=2019, agency='all'
+    )
+    assert order_results(results, 'seriesValue') == [
+        0.0, 0.0, 1.0, 1.0, 5.0, 18.0, 33.0, 34.0, 50.0, 59.0
+    ]
+
+
 def test_get_values_multiple_agencies():
     results = rc.get_values(
         series=91, jurisdiction=38, date=2019, agency=[111, 99]