Browse Source

#317 avoid pandas dependency

Falko Schindler 2 years ago
parent
commit
56100e8a9e
1 changed files with 6 additions and 4 deletions
  1. 6 4
      tests/test_plotly.py

+ 6 - 4
tests/test_plotly.py

@@ -7,16 +7,18 @@ from .screen import Screen
 
 
 def test_plotly(screen: Screen):
-    fig = px.scatter(x=[1, 2, 3], y=[1, 2, 3], title='Test')
+    fig = go.Figure(go.Scatter(x=[1, 2, 3], y=[1, 2, 3], name='Trace 1'))
+    fig.update_layout(title='Test Figure')
     plot = ui.plotly(fig)
 
     ui.button('Add trace', on_click=lambda: (
-        fig.add_trace(go.Scatter(x=[0, 1, 2], y=[2, 1, 0], name='New trace')),
+        fig.add_trace(go.Scatter(x=[0, 1, 2], y=[2, 1, 0], name='Trace 2')),
         plot.update()
     ))
 
     screen.open('/')
-    screen.should_contain('Test')
+    screen.should_contain('Test Figure')
 
     screen.click('Add trace')
-    screen.should_contain('New trace')
+    screen.should_contain('Trace 1')
+    screen.should_contain('Trace 2')