def test_sankey_data():
nodes = {}
links = {}
groups = {}
data = SankeyData(nodes, links, groups)
assert data.nodes is nodes
assert data.links is links
assert data.groups is groups
def test_sankey_data_json():
data = SankeyData(nodes=[SankeyNode(id='a')],
links=[SankeyLink(source='a', target='a')])
json = data.to_json()
assert json['nodes'] == [n.to_json() for n in data.nodes]
assert json['links'] == [l.to_json() for l in data.links]