Update test_sunburst.py

Author: Migueldfr

test/test_sunburst.py

@@ -1,4 +1,4 @@
-from sunburst import sunburst
+from toolkit.plot import sunburst
 import plotly.graph_objs as go
 import plotly.express as px
 
@@ -12,4 +12,4 @@ def test_sunburst():
 
     assert fig.layout.title.text == 'My Sunburst Chart'
     assert fig.layout.width == 800
-    assert fig.layout.height == 600
+    assert fig.layout.height == 600