mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
# 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
|