Multiple series (#17)

* test for multiple series * fixed mistaken assumption about how requests handles series
OliverSherouse · Jun 13, 2018 · 1f1b40d · 1f1b40d
1 parent 041e476
commit 1f1b40d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/bls/api.py b/bls/api.py
@@ -38,7 +38,7 @@ def unset_api_key():
 
 def _get_json_subset(series, startyear, endyear, key):
     data = {
-        "seriesid": series,
+        "seriesid": ','.join(series),
         "startyear": startyear,
         "endyear": endyear
     }

diff --git a/tests/test_api.py b/tests/test_api.py
@@ -1,6 +1,7 @@
 import datetime
 
 import bls
+import pandas as pd
 import pytest
 
 
@@ -18,6 +19,21 @@ def test_monthly_value():
     )['1948-01'] == 3.4
 
 
+def test_monthly_value_multiple():
+    assert (
+        bls.get_series(
+            ['LNS14000000', 'LNS14000001'], startyear=1948, endyear=1948
+        )
+        .xs('1948-01')
+        .equals(
+            pd.Series({
+                'LNS14000000': 3.4,
+                'LNS14000001': 3.4
+            })
+        )
+    )
+
+
 def test_quarterly_value():
     assert bls.get_series(
         'CIU2020000000000A', startyear=2001, endyear=2001