2
2
import plotly .graph_objects as go
3
3
from numpy .testing import assert_array_equal
4
4
5
+
5
6
def _compare_figures (go_trace , px_fig ):
6
7
"""Compare a figure created with a go trace and a figure created with
7
8
a px function call. Check that all values inside the go Figure are the
@@ -12,23 +13,23 @@ def _compare_figures(go_trace, px_fig):
12
13
px_fig = px_fig .to_plotly_json ()
13
14
del go_fig ["layout" ]["template" ]
14
15
del px_fig ["layout" ]["template" ]
15
- for key in go_fig [' data' ][0 ]:
16
- assert_array_equal (go_fig [' data' ][0 ][key ], px_fig [' data' ][0 ][key ])
17
- for key in go_fig [' layout' ]:
18
- assert go_fig [' layout' ][key ] == px_fig [' layout' ][key ]
16
+ for key in go_fig [" data" ][0 ]:
17
+ assert_array_equal (go_fig [" data" ][0 ][key ], px_fig [" data" ][0 ][key ])
18
+ for key in go_fig [" layout" ]:
19
+ assert go_fig [" layout" ][key ] == px_fig [" layout" ][key ]
19
20
20
21
21
22
def test_pie_like_px ():
22
23
# Pie
23
- labels = [' Oxygen' , ' Hydrogen' , ' Carbon_Dioxide' , ' Nitrogen' ]
24
+ labels = [" Oxygen" , " Hydrogen" , " Carbon_Dioxide" , " Nitrogen" ]
24
25
values = [4500 , 2500 , 1053 , 500 ]
25
26
26
27
fig = px .pie (names = labels , values = values )
27
28
trace = go .Pie (labels = labels , values = values )
28
29
_compare_figures (trace , fig )
29
30
30
31
labels = ["Eve" , "Cain" , "Seth" , "Enos" , "Noam" , "Abel" , "Awan" , "Enoch" , "Azura" ]
31
- parents = ["" , "Eve" , "Eve" , "Seth" , "Seth" , "Eve" , "Eve" , "Awan" , "Eve" ]
32
+ parents = ["" , "Eve" , "Eve" , "Seth" , "Seth" , "Eve" , "Eve" , "Awan" , "Eve" ]
32
33
values = [10 , 14 , 12 , 10 , 2 , 6 , 6 , 4 , 4 ]
33
34
# Sunburst
34
35
fig = px .sunburst (names = labels , parents = parents , values = values )
@@ -40,7 +41,7 @@ def test_pie_like_px():
40
41
_compare_figures (trace , fig )
41
42
42
43
# Funnel
43
- x = ['A' , 'B' , 'C' ]
44
+ x = ["A" , "B" , "C" ]
44
45
y = [3 , 2 , 1 ]
45
46
fig = px .funnel (y = y , x = x )
46
47
trace = go .Funnel (y = y , x = x )
@@ -49,4 +50,3 @@ def test_pie_like_px():
49
50
fig = px .funnel_area (values = y , names = x )
50
51
trace = go .Funnelarea (values = y , labels = x )
51
52
_compare_figures (trace , fig )
52
-
0 commit comments