import unittest



from pyecharts.commons import utils

from pyecharts.datasets import EXTRA





class TestUtils(unittest.TestCase):

    def test_utils_produce_require_dict(self):

        cfg = utils.produce_require_dict(

            utils.OrderedSet("echarts"), "https://example.com"

        )

        self.assertEqual(

            cfg["config_items"], ["'echarts':'https://example.comecharts.min'"]

        )

        self.assertEqual(cfg["libraries"], ["'echarts'"])



        cfg_1 = utils.produce_require_dict(

            utils.OrderedSet("https://api.map.baidu.com"),

            "https://example.com",

        )

        self.assertEqual(

            cfg_1["config_items"],

            ["'baidu_map_api25':'https://api.map.baidu.com'"],

        )

        self.assertEqual(cfg_1["libraries"], ["'baidu_map_api25'"])



        cfg_2 = utils.produce_require_dict(

            utils.OrderedSet("https://webapi.amap.com"),

            "https://example.com",

        )

        self.assertEqual(

            cfg_2["config_items"],

            ["'amap_map_api23':'https://webapi.amap.com'"],

        )

        self.assertEqual(cfg_2["libraries"], ["'amap_map_api23'"])



    def test_utils_produce_require_dict_with_extra(self):

        global EXTRA

        EXTRA["https://api.baidu.com"] = {

            "https://api.baidu.com/test.min": ["https://api.baidu.com/test.min", "css"]

        }

        cfg_0 = utils.produce_require_dict(

            utils.OrderedSet("https://api.baidu.com/test.min"),

            "https://example.com",

        )

        self.assertEqual(cfg_0["libraries"], ["'https://api.baidu.com/test.min'"])



    def test_js_code(self):

        fn = "function() { console.log('test_js_code') }"

        js_code = utils.JsCode(fn)

        self.assertEqual(js_code.js_code, "--x_x--0_0--{}--x_x--0_0--".format(fn))



    def test_ordered_set(self):

        s = utils.OrderedSet()

        s.add("a", "b", "c")

        self.assertEqual(s.items, ["a", "b", "c"])



    def test_utils_remove_key_with_none_value(self):

        mock_data = [1, 2, 3]

        list_res = utils.remove_key_with_none_value(mock_data)

        assert list_res == mock_data



        mock_data_none = None

        none_res = utils.remove_key_with_none_value(mock_data_none)

        assert none_res == mock_data_none



    def test_utils_remove_key_with_none_value_raise_value_error(self):

        import numpy as np

        import pandas as pd



        mock_data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

        mock_numpy_data = np.array(mock_data)

        tmp_df = pd.DataFrame({"x": mock_data})

        mock_series_data = tmp_df["x"]

        try:

            utils.remove_key_with_none_value({"data": mock_numpy_data})

        except ValueError:

            pass



        try:

            utils.remove_key_with_none_value({"data": mock_series_data})

        except ValueError:

            pass