def test_get_values_all_agencies():
results = rc.get_values(series=13,
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_incorrect_download(capsys):
results = rc.get_values(series=13,
jurisdiction=38,
date=2019,
agency=195,
download=True)
assert not results
assert capsys.readouterr().out == 'Valid outpath required to download.\n'
def test_get_values_multiple_industries():
results = rc.get_values(series=9,
jurisdiction=58,
date=2019,
industry=['111', '325', '326'])
assert order_results(results, 'seriesValue') == [
255.2682025779941, 649.0292048707197, 1858.660280573211
]
def test_get_values_download():
results = rc.get_values(series=13,
jurisdiction=38,
date=2019,
agency=195,
download='test.csv')
assert not results
assert os.path.exists('test.csv')
os.remove('test.csv')
def test_get_values_one_industry():
results = rc.get_values(series=9,
jurisdiction=20,
date='2020-06-02',
industry='111',
summary=False)
assert order_results(results, 'seriesValue', descending=True) == [
208.5117055773735, 177.00449323654175, 86.01110327243805,
76.680001989007, 66.70080256462097, 61.46759942173958,
58.94939732551575, 41.14539943635464, 36.53759956359863,
33.51419833302498
]
def test_get_values_all_industries():
results = rc.get_values(series=9,
jurisdiction=58,
date=2019,
industry='all',
filtered=False)
assert order_results(results, 'seriesValue') == [
16.487800191811402, 28.080800290597836, 32.408500283963804,
36.27130037093593, 36.53810011051246, 36.72170047096006,
40.113500030507566, 44.19190023323608, 45.02970027324045,
48.842899827621295
]
def test_get_values_error(capsys):
results = rc.get_values(series=1, jurisdiction=38, date=1900, verbose=1)
assert not results
assert capsys.readouterr().out == (
'WARNING: SeriesValue was not found for the specified parameters. '
'Please check that you have selected the right combination of '
'parameters. When in doubt, please use the /periods endpoint to '
'find out the combinations of series, jurisdiction, periods, '
'agencies, document types for which there are data available.'
'{parameters={jurisdiction=[US_UNITED_STATES], date=[1900], '
'industry=null, agency=null, dateIsRange=false, filteredOnly=true, '
'summary=true, series=[SERIES_1], documentType=ALL_REGULATIONS, '
'documentID=null}}\n')
def test_get_values_multiple_agencies():
results = rc.get_values(series=13,
jurisdiction=38,
date=2019,
agency=[111, 99])
assert order_results(results, 'seriesValue') == [34167.0, 91227.0]
def test_get_values_multiple_jurisdictions():
results = rc.get_values(series=1, jurisdiction=[58, 59], date=2019)
assert order_results(results, 'seriesValue') == [52569.0, 107063.0]
def test_get_values_incorrect_series(capsys):
results = rc.get_values(series=None, jurisdiction=38, date=2019)
assert not results
assert capsys.readouterr().out == (
'Valid series ID required. Select from the following list:\n')
def test_get_values_multiple_series():
results = rc.get_values(series=[1, 2],
jurisdiction=38,
date=1970,
verbose=1)
assert order_results(results, 'seriesValue') == [405647.0, 35420432.0]
def test_get_values_date_range():
results = rc.get_values(series=1, jurisdiction=38, date=[1970, 2019])
assert order_results(results, 'seriesValue') == [
405647.0, 416532.0, 452114.0, 470561.0, 500133.0, 524992.0, 548579.0,
572123.0, 581408.0, 615181.0
]
def test_get_values_multiple_dates():
results = rc.get_values(series=1,
jurisdiction=38,
date=[1970, 1980, 1990, 2000])
assert order_results(
results, 'seriesValue') == [405647.0, 633754.0, 772537.0, 853661.0]
def test_get_values_incorrect_jurisdiction(capsys):
results = rc.get_values(series=1, jurisdiction=None, date=2019)
assert not results
assert capsys.readouterr().out == 'Valid jurisdiction ID required.\n'
def test_get_values_incorrect_dates(capsys):
results = rc.get_values(series=1, jurisdiction=38, date=None)
assert not results
assert capsys.readouterr().out == 'Valid date is required.\n'
def test_get_values_agency():
results = rc.get_values(series=13, jurisdiction=38, date=2019, agency=195)
assert order_results(results, 'seriesValue') == [62.0]
def test_get_values_country():
results = rc.get_values(series=1, jurisdiction=38, date=2019, country=True)
assert order_results(results, 'seriesValue') == [
43940.0, 51925.0, 52569.0, 60086.0, 63735.0, 70969.0, 78676.0, 92522.0,
104562.0, 107063.0
]
