update test

Author: measty

tests/test_app_bokeh.py

@@ -11,19 +11,19 @@
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-import bokeh.models as bkmodels
 import matplotlib.pyplot as plt
 import numpy as np
 import pytest
 import requests
-from bokeh.application import Application
-from bokeh.application.handlers import FunctionHandler
-from bokeh.events import ButtonClick, DoubleTap, MenuItemClick
 from flask_cors import CORS
 from matplotlib import colormaps
 from PIL import Image
 from scipy.ndimage import label
 
+import bokeh.models as bkmodels
+from bokeh.application import Application
+from bokeh.application.handlers import FunctionHandler
+from bokeh.events import ButtonClick, DoubleTap, MenuItemClick
 from tiatoolbox.data import _fetch_remote_sample
 from tiatoolbox.visualization.bokeh_app import main
 from tiatoolbox.visualization.tileserver import TileServer
@@ -285,7 +285,7 @@ def test_add_annotation_layer(doc: Document, data_path: pytest.TempPathFactory)
     # trigger an event to select the geojson file
     click = MenuItemClick(layer_drop, str(data_path["geojson_anns"]))
     layer_drop._trigger_event(click)
-    assert main.UI["vstate"].types == ["annotation"]
+    assert main.UI["vstate"].types == ["nucleus", "cell", "annotation"]
 
     # test the name2type function.
     assert main.name2type("annotation") == '"annotation"'