# coding=utf8
"""
Test cases for rendering in jupyter notebook
"""
from __future__ import unicode_literals

import json

from pyecharts import Bar, Line, Pie, Page, online
from pyecharts import enable_nteract, configure
import pyecharts.constants as constants
from pyecharts.conf import CURRENT_CONFIG
from test.constants import CLOTHES, WEEK

TITLE = "柱状图数据堆叠示例"


def create_a_bar(title):
 v1 = [5, 20, 36, 10, 75, 90]
 v2 = [10, 25, 8, 60, 20, 80]
 bar = Bar(title)
 bar.add("商家A", CLOTHES, v1, is_stack=True)
 bar.add("商家B", CLOTHES, v2, is_stack=True)
 return bar


def test_single_chart():
 bar = create_a_bar(TITLE)
 html = bar._repr_html_()
 json_encoded_title = json.dumps(TITLE)
 assert json_encoded_title in html
 assert "require.config" in html
 assert "function(echarts)" in html
 assert "nbextensions/echarts" in html
 assert "width:800px" in html
 assert "height:400px" in html
 assert "<div" in html
 assert "myChart" in html
 assert bar.chart_id in html


def test_page():
 page = Page()
 line = Line("折线图示例")
 line.chart_id = "id_my_cell_line"
 line.add(
 "最高气温",
 WEEK,
 [11, 11, 15, 13, 12, 13, 10],
 mark_point=["max", "min"],
 mark_line=["average"],
 )
 line.add(
 "最低气温",
 WEEK,
 [1, -2, 2, 5, 3, 2, 0],
 mark_point=["max", "min"],
 mark_line=["average"],
 )

 # pie
 v1 = [11, 12, 13, 10, 10, 10]
 pie = Pie("饼图-圆环图示例", title_pos="center", width="600px")
 pie.add(
 "",
 CLOTHES,
 v1,
 radius=[40, 75],
 label_text_color=None,
 is_label_show=True,
 legend_orient="vertical",
 legend_pos="left",
 )

 page.add([line, pie, create_a_bar(TITLE)])
 # Start render and test
 html = page._repr_html_()
 # Test base html structure
 assert html.count("<script>") == html.count("</script>") == 2
 assert html.count("<div") == html.count("</div>") == 3
 assert html.count("require.config") == html.count("function(echarts)") == 1
 # Test some chart attributes
 json_encoded_title = json.dumps(TITLE)
 assert json_encoded_title in html
 assert "nbextensions/echarts" in html # default jshost
 assert html.count("height:400px") == 3
 assert html.count("width:600px") == 1
 assert html.count("width:800px") == 2
 assert html.count("id_my_cell_line") == 6


def test_online_feature():
 online()
 bar = create_a_bar(TITLE)
 html = bar._repr_html_()
 expected_jshost = "https://pyecharts.github.io/jupyter-echarts/echarts"
 assert expected_jshost in html
 CURRENT_CONFIG.hosted_on_github = False


def test_online_with_custom_jshost():
 online(host="https://my-site.com/js")
 bar = create_a_bar(TITLE)
 html = bar._repr_html_()
 expected_jshost = "https://my-site.com/js"
 assert expected_jshost in html
 CURRENT_CONFIG.jshost = None


def test_nteract_feature():
 enable_nteract()
 bar = create_a_bar(TITLE)
 html = bar._repr_html_()
 assert "https://pyecharts.github.io/assets/js/echarts.min.js" in html
 assert "require" not in html
 # restore configuration
 configure(output_image=constants.DEFAULT_HTML)
 CURRENT_CONFIG.jshost = None