Fix test_grouped_callbacks

plotly · Jun 13, 2022 · c5f6fff · c5f6fff
1 parent 2b22ff3
commit c5f6fff
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/unit/dash/test_grouped_callbacks.py b/tests/unit/dash/test_grouped_callbacks.py
@@ -1,6 +1,7 @@
 import dash
 from dash._grouping import make_grouping_by_index, grouping_len, flatten_grouping
 from dash._utils import create_callback_id
+from dash._callback import GLOBAL_CALLBACK_MAP
 from dash.dependencies import Input, State, Output, ClientsideFunction
 import mock
 import json
@@ -50,7 +51,7 @@ def check_output_for_grouping(grouping):
         Input("input-a", "prop"),
     )(mock_fn)
 
-    wrapped_fn = app.callback_map[callback_id]["callback"]
+    wrapped_fn = GLOBAL_CALLBACK_MAP[callback_id]["callback"]
 
     expected_outputs = [
         (dep.component_id, dep.component_property, val)
@@ -108,7 +109,7 @@ def check_callback_inputs_for_grouping(grouping):
         inputs,
     )(mock_fn)
 
-    wrapped_fn = app.callback_map["output-a.prop"]["callback"]
+    wrapped_fn = GLOBAL_CALLBACK_MAP["output-a.prop"]["callback"]
 
     flat_input_state_values = flatten_grouping(grouping)
     flat_input_values = flat_input_state_values[0::2]