def test_geo_visualmap_pieces():
data = [
("海门", 9),
("鄂尔多斯", 12),
("招远", 12),
("舟山", 12),
("齐齐哈尔", 14),
("盐城", 15),
]
geo = Geo("全国主要城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(data)
geo.add(
"",
attr,
value,
type="effectScatter",
is_random=True,
is_visualmap=True,
is_piecewise=True,
visual_text_color="#fff",
pieces=[
{"min": 0, "max": 13, "label": "0 < x < 13"},
{"min": 14, "max": 16, "label": "14 < x < 16"},
],
effect_scale=5,
)
content = geo._repr_html_()
assert '"max": 13' in content
assert '"label": "14 < x < 16"' in content
def test_geo_type_effectscatter():
data = [
("海门", 9), ("鄂尔多斯", 12), ("招远", 12),
("舟山", 12), ("齐齐哈尔", 14), ("盐城", 15)
]
geo = Geo("全国主要城市空气质量", "data from pm2.5", **chart_init)
attr, value = geo.cast(data)
geo.add("", attr, value, type="effectScatter", is_random=True,
effect_scale=5)
assert '"type": "effectScatter"' in geo._repr_html_()
def test_geo_china_effectscatter():
data = [
("海门", 9), ("鄂尔多斯", 12), ("招远", 12),
("舟山", 12), ("齐齐哈尔", 14), ("盐城", 15)
]
geo = Geo("全国主要城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(data)
geo.add("", attr, value, type="effectScatter", is_random=True,
effect_scale=5)
assert '"type": "effectScatter"' in geo._repr_html_()
def test_geo_china_scatter():
geo = Geo("全国主要城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(cities)
geo.add("", attr, value, visual_range=[0, 200],
visual_text_color="#fff", is_legend_show=False,
symbol_size=15, is_visualmap=True, tooltip_formatter='{b}',
label_emphasis_textsize=15, label_emphasis_pos='right')
html_content = geo._repr_html_()
assert '"type": "scatter"' in html_content
assert '"type": "heatmap"' not in html_content
assert '"type": "effectScatter"' not in html_content
def test_geo_china_heatmap():
geo = Geo("全国主要城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(cities)
geo.add("",
attr,
value,
type="heatmap",
is_visualmap=True,
visual_range=[0, 300],
visual_text_color='#fff')
assert '"type": "heatmap"' in geo._repr_html_()
def test_geo_china_heatmap():
geo = Geo("全国主要城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(cities)
geo.add(
"",
attr,
value,
type="heatmap",
is_visualmap=True,
visual_range=[0, 300],
visual_text_color="#fff",
)
assert '"type": "heatmap"' in geo._repr_html_()
def test_geo_formatter_func():
style = Style(
title_color="#fff",
title_pos="center",
width=1200,
height=600,
background_color="#404a59",
)
data = [("汕头市", 50), ("汕尾市", 60), ("揭阳市", 35), ("阳江市", 44), ("肇庆市", 72)]
geo = Geo("广东城市空气质量", "data from pm2.5", **style.init_style)
attr, value = geo.cast(data)
geo.add(
"",
attr,
value,
maptype="广东",
type="effectScatter",
tooltip_formatter=geo_formatter,
is_random=True,
effect_scale=5,
is_legend_show=False,
)
assert "function geo_formatter(" in geo._repr_html_()
def test_geo_china():
data = [
("海门", 9),
("鄂尔多斯", 12),
("招远", 12),
("舟山", 12),
("齐齐哈尔", 14),
("盐城", 15),
("赤峰", 16),
("青岛", 18),
("乳山", 18),
("金昌", 19),
("泉州", 21),
("莱西", 21),
("日照", 21),
("胶南", 22),
("南通", 23),
("拉萨", 24),
("云浮", 24),
("梅州", 25),
("文登", 25),
("上海", 25),
("攀枝花", 25),
("威海", 25),
("承德", 25),
("厦门", 26),
("汕尾", 26),
("潮州", 26),
("丹东", 27),
("太仓", 27),
("曲靖", 27),
("烟台", 28),
("福州", 29),
("瓦房店", 30),
("即墨", 30),
("抚顺", 31),
("玉溪", 31),
("张家口", 31),
("阳泉", 31),
("莱州", 32),
("湖州", 32),
("汕头", 32),
("昆山", 33),
("宁波", 33),
("湛江", 33),
("揭阳", 34),
("荣成", 34),
("连云港", 35),
("葫芦岛", 35),
("常熟", 36),
("东莞", 36),
("河源", 36),
("淮安", 36),
("泰州", 36),
("南宁", 37),
("营口", 37),
("惠州", 37),
("江阴", 37),
("蓬莱", 37),
("韶关", 38),
("嘉峪关", 38),
("广州", 38),
("延安", 38),
("太原", 39),
("清远", 39),
("中山", 39),
("昆明", 39),
("寿光", 40),
("盘锦", 40),
("长治", 41),
("深圳", 41),
("珠海", 42),
("宿迁", 43),
("咸阳", 43),
("铜川", 44),
("平度", 44),
("佛山", 44),
("海口", 44),
("江门", 45),
("章丘", 45),
("肇庆", 46),
("大连", 47),
("临汾", 47),
("吴江", 47),
("石嘴山", 49),
("沈阳", 50),
("苏州", 50),
("茂名", 50),
("嘉兴", 51),
("长春", 51),
("胶州", 52),
("银川", 52),
("张家港", 52),
("三门峡", 53),
("锦州", 54),
("南昌", 54),
("柳州", 54),
("三亚", 54),
("自贡", 56),
("吉林", 56),
("阳江", 57),
("泸州", 57),
("西宁", 57),
("宜宾", 58),
("呼和浩特", 58),
("成都", 58),
("大同", 58),
("镇江", 59),
("桂林", 59),
("张家界", 59),
("宜兴", 59),
("北海", 60),
("西安", 61),
("金坛", 62),
("东营", 62),
("牡丹江", 63),
("遵义", 63),
("绍兴", 63),
("扬州", 64),
("常州", 64),
("潍坊", 65),
("重庆", 66),
("台州", 67),
("南京", 67),
("滨州", 70),
("贵阳", 71),
("无锡", 71),
("本溪", 71),
("克拉玛依", 72),
("渭南", 72),
("马鞍山", 72),
("宝鸡", 72),
("焦作", 75),
("句容", 75),
("北京", 79),
("徐州", 79),
("衡水", 80),
("包头", 80),
("绵阳", 80),
("乌鲁木齐", 84),
("枣庄", 84),
("杭州", 84),
("淄博", 85),
("鞍山", 86),
("溧阳", 86),
("库尔勒", 86),
("安阳", 90),
("开封", 90),
("济南", 92),
("德阳", 93),
("温州", 95),
("九江", 96),
("邯郸", 98),
("临安", 99),
("兰州", 99),
("沧州", 100),
("临沂", 103),
("南充", 104),
("天津", 105),
("富阳", 106),
("泰安", 112),
("诸暨", 112),
("郑州", 113),
("哈尔滨", 114),
("聊城", 116),
("芜湖", 117),
("唐山", 119),
("平顶山", 119),
("邢台", 119),
("德州", 120),
("济宁", 120),
("荆州", 127),
("宜昌", 130),
("义乌", 132),
("丽水", 133),
("洛阳", 134),
("秦皇岛", 136),
("株洲", 143),
("石家庄", 147),
("莱芜", 148),
("常德", 152),
("保定", 153),
("湘潭", 154),
("金华", 157),
("岳阳", 169),
("长沙", 175),
("衢州", 177),
("廊坊", 193),
("菏泽", 194),
("合肥", 229),
("武汉", 273),
("大庆", 279)
]
# geo type 'scatter'
geo = Geo("全国主要城市空气质量", "data from pm2.5", **chart_init)
attr, value = geo.cast(data)
geo.add("", attr, value, visual_range=[0, 200],
visual_text_color="#fff", is_legend_show=False,
symbol_size=15, is_visualmap=True, tooltip_formatter='{b}',
label_emphasis_textsize=15, label_emphasis_pos='right')
html_content = geo._repr_html_()
assert '"type": "scatter"' in html_content
assert '"type": "heatmap"' not in html_content
assert '"type": "effectScatter"' not in html_content
# geo type 'heatmap'
geo = Geo("全国主要城市空气质量", "data from pm2.5", **chart_init)
attr, value = geo.cast(data)
geo.add("", attr, value, type="heatmap", is_visualmap=True,
visual_range=[0, 300], visual_text_color='#fff')
assert '"type": "heatmap"' in geo._repr_html_()
