def test_display_vector_extruded_ChoroplethViz(display):
"""Assert that show calls the mocked display function when using data-join technique
for ChoroplethViz.
"""
data = [{
"id": "06",
"name": "California",
"density": 241.7
}, {
"id": "11",
"name": "District of Columbia",
"density": 10065
}, {
"id": "25",
"name": "Massachusetts",
"density": 840.2
}, {
"id": "30",
"name": "Montana",
"density": 6.858
}, {
"id": "36",
"name": "New York",
"density": 412.3
}, {
"id": "49",
"name": "Utah",
"density": 34.3
}, {
"id": "72",
"name": "Puerto Rico",
"density": 1082
}]
viz = ChoroplethViz(data,
vector_url='mapbox://mapbox.us_census_states_2015',
vector_layer_name='states',
vector_join_property='STATEFP',
data_join_property='id',
color_property='density',
color_stops=create_color_stops([0, 50, 100, 500, 1500],
colors='YlOrRd'),
height_property='density',
height_stops=create_numeric_stops(
[0, 50, 100, 500, 1500, 10000], 0, 1000000),
access_token=TOKEN)
viz.show()
display.assert_called_once()
def test_display_vector_LinestringViz(display):
"""Assert that show calls the mocked display function when using data-join technique
for LinestringViz.
"""
data = [{
"elevation": x,
"weight": random.randint(0, 100)
} for x in range(0, 21000, 10)]
viz = LinestringViz(data,
vector_url='mapbox://mapbox.mapbox-terrain-v2',
vector_layer_name='contour',
vector_join_property='ele',
data_join_property='elevation',
color_property="elevation",
color_stops=create_color_stops([0, 50, 100, 500, 1500],
colors='YlOrRd'),
line_width_property='weight',
line_width_stops=create_numeric_stops(
[0, 25, 50, 75, 100], 1, 6),
access_token=TOKEN)
viz.show()
display.assert_called_once()
def test_create_numeric_stops():
"""Create numeric stops from custom breaks"""
domain = [7678.214347826088, 5793.63142857143, 1200]
stops = create_numeric_stops(domain, 1, 10)
assert stops == [[7678.214347826088, 1.0], [5793.63142857143, 4.0],
[1200, 7.0]]