2727import  pyvista  as  pv 
2828from  pyvista .plotting .plotter  import  Plotter  as  PyVistaPlotter 
2929
30- from  ansys .tools .visualization_interface  import   DOCUMENTATION_BUILD ,  TESTING_MODE 
30+ import  ansys .tools .visualization_interface  as   viz_interface 
3131from  ansys .tools .visualization_interface .types .edge_plot  import  EdgePlot 
3232from  ansys .tools .visualization_interface .types .mesh_object_plot  import  MeshObjectPlot 
3333from  ansys .tools .visualization_interface .utils .clip_plane  import  ClipPlane 
@@ -74,7 +74,7 @@ def __init__(
7474        # Generate custom scene if ``None`` is provided 
7575        if  scene  is  None :
7676            scene  =  pv .Plotter (plotter_kwargs )
77-         if  TESTING_MODE :
77+         if  viz_interface . TESTING_MODE :
7878            scene .off_screen  =  True 
7979        # If required, use a white background with no gradient 
8080        if  not  color_opts :
@@ -326,7 +326,7 @@ def show(
326326            jupyter_backend  =  "trame" 
327327
328328        # Override Jupyter backend if building docs 
329-         if  DOCUMENTATION_BUILD :
329+         if  viz_interface . DOCUMENTATION_BUILD :
330330            jupyter_backend  =  "html" 
331331
332332        # Enabling anti-aliasing by default on scene 
@@ -337,7 +337,7 @@ def show(
337337            self .scene .off_screen  =  True 
338338
339339        # If running on testing, set off_screen to True for the plotter 
340-         if  TESTING_MODE  or  DOCUMENTATION_BUILD :
340+         if  viz_interface . TESTING_MODE  or  viz_interface . DOCUMENTATION_BUILD :
341341            self .scene .off_screen  =  True 
342342
343343        self .scene .show (jupyter_backend = jupyter_backend , ** kwargs )
0 commit comments