def test_themeriver_basefake_writer(fake_writer): data = [ ["2015/11/08", 10, "DQ"], ["2015/11/20", 30, "TY"], ["2015/11/08", 21, "SS"], ["2015/11/14", 7, "QG"], ["2015/11/22", 4, "SY"], ["2015/11/20", 26, "DD"], ] c = ThemeRiver().add( ["DQ", "TY", "SS", "QG", "SY", "DD"], data, singleaxis_opts=opts.SingleAxisOpts(type_="time", pos_bottom="10%"), ) c.render() _, content = fake_writer.call_args[0] eq_(c.theme, "white") eq_(c.renderer, "canvas")
def draw_river_picture(series_name, data, to_file, svg_name): """ 主题河流图 :param series_name: [str] :param data: [[date, value, series_name]] :param svg_name: svg文件 :param to_file: 结果文件 :return: """ bar = ThemeRiver(init_opts=opts.InitOpts(width="800px", height="600px", bg_color='white')) \ .add( series_name=series_name, data=data, singleaxis_opts=opts.SingleAxisOpts( pos_top="2%", pos_bottom="10%", pos_right="20%", type_="time", name_textstyle_opts=opts.TextStyleOpts(font_size=25), axislabel_opts=opts.LabelOpts(font_size=25) ), label_opts=opts.LabelOpts(is_show=False, position='bottom', distance='200px'), ) \ .set_global_opts( tooltip_opts=opts.TooltipOpts(trigger="axis", axis_pointer_type="line"), toolbox_opts=opts.ToolboxOpts( feature=opts.ToolBoxFeatureOpts( # data_zoom=opts.ToolBoxFeatureDataZoomOpts(is_show=False), # brush=opts.ToolBoxFeatureBrushOpts(is_show=False), ) ), legend_opts=opts.LegendOpts(type_="scroll", pos_left="82%", pos_top="18%", orient="vertical", backgroundColor='rgb(255, 255, 255)', item_width=40, item_height=20, textstyle_opts=opts.TextStyleOpts(font_size=25) ) ) make_snapshot(snapshot, bar.render(to_file), svg_name) # 生成svg图片