def test_update_lists_get_street_data():
"""Check modify in place."""
# Initialize lists for street data
key_list = []
name_list = []
lon_list = []
lat_list = []
speed_list = []
road_list = []
nbhd_list = []
# Get street data
idx2poly = get_polygons(SHP_PATH)
data_frame = gpd.read_file(get_flow_path(2007))
get_street_data(data_frame, 2007, idx2poly, key_list, name_list, lon_list,
lat_list, speed_list, road_list, nbhd_list)
# Check length
assert len(key_list) > 0
assert len(name_list) > 0
assert len(lon_list) > 0
assert len(lat_list) > 0
assert len(speed_list) > 0
assert len(road_list) > 0
assert len(nbhd_list) > 0
def test_one_nbhd_get_neighborhood():
"""Check output of one neighborhood."""
lon = [-122.36866107043400]
lat = [47.66757206792280]
idx2poly = get_polygons(SHP_PATH)
nbhd_list = get_neighborhood(lon, lat, idx2poly)
assert len(nbhd_list) == 1
assert nbhd_list[0] == 0
def test_output_add_flow_data():
"""Check final DataFrame."""
# Get neighborhood polygons
idx2poly = get_polygons(SHP_PATH)
# Get mapping from FLOWSEGID to COMPKEY
flow2key, _, df_list, year_list = get_flow_data()
# Initialize lists for street data
key_list = []
name_list = []
lon_list = []
lat_list = []
speed_list = []
road_list = []
nbhd_list = []
# Get street data from Seattle Streets dataset
data_frame = gpd.read_file(STREET_PATH)
get_street_data(data_frame, 'street', idx2poly, key_list, name_list,
lon_list, lat_list, speed_list, road_list, nbhd_list)
# Get street data from Traffic Flow Counts datasets
for i in range(len(df_list) - 1, -1, -1):
get_street_data(df_list[i], year_list[i], idx2poly, key_list,
name_list, lon_list, lat_list, speed_list, road_list,
nbhd_list)
# Create initial DataFrame
df_streets = pd.DataFrame(
data={
'key': key_list,
'name': name_list,
'lon': lon_list,
'lat': lat_list,
'speed': speed_list,
'road': road_list,
'nbhd': nbhd_list
})
# Add traffic flow data
for i in range(len(df_list) - 1, -1, -1):
add_flow_data(df_streets, df_list[i], year_list[i], flow2key)
# Check column names and types
name_list = ['key', 'name', 'lon', 'lat', 'speed', 'road', 'nbhd']
type_list = [int, object, list, list, int, int, list]
for i in range(7):
assert list(df_streets)[i] == name_list[i]
assert df_streets.dtypes[i] == type_list[i]
for year in range(2007, 2019):
assert list(df_streets)[year - 2000] == str(year)
assert df_streets.dtypes[year - 2000] == int
# Check values
for year in range(2007, 2019):
assert max(df_streets[str(year)].to_list()) > -1
assert max(df_streets[str(year)].to_list()) > -1
def test_two_nbhds_neighborhood():
"""Check output of two neighborhoods."""
lon = [-122.36866107043400, -122.3821835817560]
lat = [47.66757206792280, 47.69606176398850]
idx2poly = get_polygons(SHP_PATH)
nbhd_list = get_neighborhood(lon, lat, idx2poly)
assert len(nbhd_list) == 2
assert nbhd_list[0] == 0
assert nbhd_list[1] == 1
def test_output_type_get_neighborhood():
"""Check output type."""
lon = [-122.36866107043400]
lat = [47.66757206792280]
idx2poly = get_polygons(SHP_PATH)
nbhd_list = get_neighborhood(lon, lat, idx2poly)
nbhd_type = set(type(n) for n in nbhd_list)
assert isinstance(nbhd_list, list)
assert len(nbhd_type) == 1
assert isinstance(nbhd_list[0], int)
def test_output_types_get_street_data():
"""Check output types."""
# Initialize lists for street data
key_list = []
name_list = []
lon_list = []
lat_list = []
speed_list = []
road_list = []
nbhd_list = []
# Get street data
idx2poly = get_polygons(SHP_PATH)
data_frame = gpd.read_file(get_flow_path(2007))
get_street_data(data_frame, 2007, idx2poly, key_list, name_list, lon_list,
lat_list, speed_list, road_list, nbhd_list)
# Get types
key_type = set(type(k) for k in key_list)
name_type = set(type(n) for n in name_list)
lon_type = set(type(l) for l in lon_list)
lat_type = set(type(l) for l in lat_list)
speed_type = set(type(s) for s in speed_list)
road_type = set(type(r) for r in road_list)
nbhd_type = set(type(n) for n in nbhd_list)
# Check number of types
assert len(key_type) == 1
assert len(name_type) == 1
assert len(lon_type) == 1
assert len(lat_type) == 1
assert len(speed_type) == 1
assert len(road_type) == 1
assert len(nbhd_type) == 1
# Check types
assert list(key_type)[0] == int
assert list(name_type)[0] == str
assert list(lon_type)[0] == list
assert list(lat_type)[0] == list
assert list(speed_type)[0] == int
assert list(road_type)[0] == int
assert list(nbhd_type)[0] == list
def test_output_range_get_street_data():
"""Check range of output."""
# Initialize lists for street data
key_list = []
name_list = []
lon_list = []
lat_list = []
speed_list = []
road_list = []
nbhd_list = []
# Get street data
idx2poly = get_polygons(SHP_PATH)
data_frame = gpd.read_file(STREET_PATH)
get_street_data(data_frame, 'street', idx2poly, key_list, name_list,
lon_list, lat_list, speed_list, road_list, nbhd_list)
# Check range
assert min(speed_list) >= -1
assert max(speed_list) <= 60
assert min(road_list) >= 0
assert max(road_list) <= 5
def test_output_length_get_polygons():
"""Check length and output types."""
idx2poly = get_polygons(SHP_PATH)
assert len(idx2poly) == 103
assert isinstance(idx2poly, dict)
assert isinstance(idx2poly[0], Polygon)
def test_exception_get_polygons():
"""Check that function throws an error if no file at shp_path."""
with pytest.raises(Exception):
assert get_polygons('dummy.shp')