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 ]