pyecharts/test/test_tree.py

# coding=utf-8
from __future__ import unicode_literals

import codecs
import json
import os
from copy import deepcopy

from pyecharts import Tree


def test_tree_default():
    data = [
        {
            "children": [
                {"children": [], "name": "B"},
                {
                    "children": [
                        {
                            "children": [{"children": [], "name": "I"}],
                            "name": "E",
                        },
                        {"children": [], "name": "F"},
                    ],
                    "name": "C",
                },
                {
                    "children": [
                        {
                            "children": [
                                {"children": [], "name": "J"},
                                {"children": [], "name": "K"},
                            ],
                            "name": "G",
                        },
                        {"children": [], "name": "H"},
                    ],
                    "name": "D",
                },
            ],
            "name": "A",
        }
    ]
    copy_data = deepcopy(data)
    data1 = Tree._set_collapse_interval(data)
    assert copy_data == data1


def test_tree_collapse_interval():
    with codecs.open(
        os.path.join("fixtures", "flare.json"), "r", encoding="utf-8"
    ) as f:
        j = json.load(f)
    copy_data = deepcopy(j)
    data = Tree._set_collapse_interval([j], interval=2)
    assert data != copy_data