test correction

holoviz · Apr 2, 2019 · ad5b899 · ad5b899
1 parent 4d98ac8
commit ad5b899
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/panel/pane/tests/test_ace.py b/panel/pane/tests/test_ace.py
@@ -1,6 +1,6 @@
 from panel.pane import Ace
 
-code="""
+code = """
 import math
 x = 10
 math.sin(x)**2 + math.cos(x)**2 == 1
@@ -11,8 +11,9 @@ def test_ace_pane(document, comm):
     pane = Ace(code=code)
 
     # Create pane
-    model = pane.get_root(document, comm=comm)
-    assert pane._models[model.ref['id']][0] is model
+    root = pane.get_root(document, comm=comm)
+    model = pane._get_model(document, root, None, comm)
+    assert pane._models[root.ref['id']][0] is model
     assert model.code == code
 
     # Cleanup