Function setup_data

tests/test_folium.py:28–42

Import economic data for testing.

()

def setup_data():
    """Import economic data for testing."""
    with open(os.path.join(rootpath, "us-counties.json")) as f:
        get_id = json.load(f)

    county_codes = [x["id"] for x in get_id["features"]]
    county_df = pd.DataFrame({"FIPS_Code": county_codes}, dtype=str)

    # Read into Dataframe, cast to string for consistency.
    df = pd.read_csv(os.path.join(rootpath, "us_county_data.csv"), na_values=[" "])
    df["FIPS_Code"] = df["FIPS_Code"].astype(str)

    # Perform an inner join, pad NA's with data from nearest county.
    merged = pd.merge(df, county_df, on="FIPS_Code", how="inner")
    return merged.fillna(method="pad")


def test_location_args():