projektAI/venv/Lib/site-packages/matplotlib/tests/test_sankey.py

25 lines
670 B
Python
Raw Normal View History

2021-06-06 22:13:05 +02:00
from matplotlib.sankey import Sankey
def test_sankey():
# lets just create a sankey instance and check the code runs
sankey = Sankey()
sankey.add()
def test_label():
s = Sankey(flows=[0.25], labels=['First'], orientations=[-1])
assert s.diagrams[0].texts[0].get_text() == 'First\n0.25'
def test_format_using_callable():
# test using callable by slightly incrementing above label example
def show_three_decimal_places(value):
return f'{value:.3f}'
s = Sankey(flows=[0.25], labels=['First'], orientations=[-1],
format=show_three_decimal_places)
assert s.diagrams[0].texts[0].get_text() == 'First\n0.250'