def test_invalid_file(self): with open(os.getcwd() + "/" + "sample.txt", "w+") as file: file.write( "395133768, 10212 BUCKEYE RD, Cleveland, Ohio, (41.48195701269418, -81.6138601319609)" ) api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) loc = fs.location.get_detail("sample.txt", "property", csv=True) assert len(loc) == 1 os.remove(os.getcwd() + "/" + "sample.txt")
def test_multi_type(self): api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) loc = fs.location.get_detail([ 395133768, "10212 BUCKEYE RD, Cleveland, Ohio", (41.48195701269418, -81.6138601319609) ], "property", csv=True) assert len(loc) == 3 assert loc[0].route == 'BUCKEYE RD' assert loc[1].route == 'BUCKEYE RD' assert loc[2].route == 'BUCKEYE RD'
# Author: Kelvin Lai <*****@*****.**> # Copyright: This module is owned by First Street Foundation # Standard Imports import os # External Imports import pytest # Internal Imports import firststreet from firststreet.errors import InvalidArgument api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) class TestEnvironmentalEvent: def test_empty(self): with pytest.raises(InvalidArgument): fs.environmental.get_precipitation([], "") def test_wrong_fsid_type(self): with pytest.raises(InvalidArgument): fs.environmental.get_precipitation(19117) def test_invalid(self): fsid = [0000] environmental = fs.environmental.get_precipitation(fsid) assert len(environmental) == 1 assert environmental[0].fsid == str(fsid[0])
if not argument.api_key: env_var_name = 'FSF_API_KEY' try: api_key = os.environ[env_var_name] except KeyError: logging.error("`{}` is not set as an Environmental Variable".format(env_var_name)) sys.exit() else: api_key = argument.api_key limit = int(argument.connection_limit) rate_limit = int(argument.rate_limit) rate_period = int(argument.rate_period) fs = firststreet.FirstStreet(api_key, version=argument.version, connection_limit=limit, rate_limit=rate_limit, rate_period=rate_period, log=bool(strtobool(argument.log))) if argument.product == 'adaptation.get_detail': fs.adaptation.get_detail(search_items, csv=True, extra_param=argument.extra_param) elif argument.product == 'adaptation.get_summary': fs.adaptation.get_summary(search_items, argument.location, csv=True, extra_param=argument.extra_param) elif argument.product == 'adaptation.get_details_by_location': fs.adaptation.get_details_by_location(search_items,
def test_full(tmpdir): api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) fs.adaptation.get_detail([29], csv=True, output_dir=tmpdir) fs.adaptation.get_summary([395133768], "property", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([7924], "neighborhood", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([1935265], "city", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([50158], "zcta", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([39061007100], "tract", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([19047], "county", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([3915], "cd", csv=True, output_dir=tmpdir) fs.adaptation.get_summary([39], "state", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([395133768], "property", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([7924], "neighborhood", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([1935265], "city", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([50158], "zcta", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([39061007100], "tract", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([19047], "county", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([3915], "cd", csv=True, output_dir=tmpdir) fs.adaptation.get_details_by_location([39], "state", csv=True, output_dir=tmpdir) fs.probability.get_chance([390000257], csv=True, output_dir=tmpdir) fs.probability.get_count([7935], 'neighborhood', csv=True, output_dir=tmpdir) fs.probability.get_count([1959835], 'city', csv=True, output_dir=tmpdir) fs.probability.get_count([44203], 'zcta', csv=True, output_dir=tmpdir) fs.probability.get_count([39035103400], 'tract', csv=True, output_dir=tmpdir) fs.probability.get_count([39047], 'county', csv=True, output_dir=tmpdir) fs.probability.get_count([3904], 'cd', csv=True, output_dir=tmpdir) fs.probability.get_count([39], 'state', csv=True, output_dir=tmpdir) fs.probability.get_count_summary([394406220], csv=True, output_dir=tmpdir) fs.probability.get_cumulative([390000439], csv=True, output_dir=tmpdir) fs.probability.get_depth([390000227], csv=True, output_dir=tmpdir) fs.environmental.get_precipitation([39057], csv=True, output_dir=tmpdir) fs.historic.get_event([2], csv=True, output_dir=tmpdir) fs.historic.get_summary([511447411], "property", csv=True, output_dir=tmpdir) fs.historic.get_summary([540225], "neighborhood", csv=True, output_dir=tmpdir) fs.historic.get_summary([1982200], "city", csv=True, output_dir=tmpdir) fs.historic.get_summary([50156], "zcta", csv=True, output_dir=tmpdir) fs.historic.get_summary([19153004900], "tract", csv=True, output_dir=tmpdir) fs.historic.get_summary([19163], "county", csv=True, output_dir=tmpdir) fs.historic.get_summary([1901], "cd", csv=True, output_dir=tmpdir) fs.historic.get_summary([39], "state", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([511447411], "property", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([540225], "neighborhood", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([1982200], "city", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([50156], "zcta", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([19153004900], "tract", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([19163], "county", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([1901], "cd", csv=True, output_dir=tmpdir) fs.historic.get_events_by_location([39], "state", csv=True, output_dir=tmpdir) fs.location.get_detail([511447411], "property", csv=True, output_dir=tmpdir) fs.location.get_detail([1206631], "neighborhood", csv=True, output_dir=tmpdir) fs.location.get_detail([3915406], "city", csv=True, output_dir=tmpdir) fs.location.get_detail([44654], "zcta", csv=True, output_dir=tmpdir) fs.location.get_detail([39151712602], "tract", csv=True, output_dir=tmpdir) fs.location.get_detail([39077], "county", csv=True, output_dir=tmpdir) fs.location.get_detail([3904], "cd", csv=True, output_dir=tmpdir) fs.location.get_detail([39], "state", csv=True, output_dir=tmpdir) fs.location.get_summary([395112095], "property", csv=True, output_dir=tmpdir) fs.location.get_summary([631054], "neighborhood", csv=True, output_dir=tmpdir) fs.location.get_summary([3958002], "city", csv=True, output_dir=tmpdir) fs.location.get_summary([43935], "zcta", csv=True, output_dir=tmpdir) fs.location.get_summary([39153531702], "tract", csv=True, output_dir=tmpdir) fs.location.get_summary([39027], "county", csv=True, output_dir=tmpdir) fs.location.get_summary([3903], "cd", csv=True, output_dir=tmpdir) fs.location.get_summary([39], "state", csv=True, output_dir=tmpdir) fs.fema.get_nfip([44074], "zcta", csv=True, output_dir=tmpdir) fs.fema.get_nfip([39013012300], "tract", csv=True, output_dir=tmpdir) fs.fema.get_nfip([39093], "county", csv=True, output_dir=tmpdir) fs.fema.get_nfip([39], "state", csv=True, output_dir=tmpdir)
def test_invalid_call(self): api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) with pytest.raises(InvalidArgument): fs.location.get_detail([392873515], "", csv=True)
def test_valid_call(self): api_key = os.environ['FSF_API_KEY'] fs = firststreet.FirstStreet(api_key) adap = fs.adaptation.get_detail([29], csv=False, connection_limit=100) assert len(adap) == 1 assert adap[0].name == 'Riverfront Park'
def test_vaid_key(self): api_key = os.environ['FSF_API_KEY'] firststreet.FirstStreet(api_key)
def test_invalid_key(self): with pytest.raises(MissingAPIKeyError): firststreet.FirstStreet("")
# Try to get the API key either from env var or the parameter if not argument.api_key: env_var_name = 'FSF_API_KEY' try: api_key = os.environ[env_var_name] except KeyError: logging.error( "`{}` is not set as an Environmental Variable".format( env_var_name)) sys.exit() else: api_key = argument.api_key fs = firststreet.FirstStreet(api_key, version=argument.version, log=bool(strtobool(argument.log))) limit = int(argument.connection_limit) if argument.product == 'adaptation.get_detail': fs.adaptation.get_detail(search_items, csv=True, connection_limit=limit, extra_param=argument.extra_param) elif argument.product == 'adaptation.get_summary': fs.adaptation.get_summary(search_items, argument.location, csv=True, connection_limit=limit,