WIP: fixing regressions in tests

tests/test_auto_slices.py

@@ -50,7 +50,8 @@ def test_add_selected_triggers_auto_slices(mock_app):
  mock_app.files = [f_mock]
  mock_app._ui_picker_selected = {"test.py"}
  
- with patch('src.gui_2.imgui') as mock_imgui:
+ with patch('src.gui_2.imgui') as mock_imgui, \
+      patch('src.gui_2.imscope') as mock_imscope:
   mock_imgui.begin_popup_modal.return_value = (True, True)
   mock_imgui.button.side_effect = lambda label, size=None: label == "Add Selected"
   mock_imgui.checkbox.return_value = (False, False)
@@ -58,6 +59,12 @@ def test_add_selected_triggers_auto_slices(mock_app):
   mock_imgui.ImVec2 = MagicMock()
   mock_imgui.WindowFlags_ = MagicMock()
   
+  # Setup imscope mocks to act as context managers
+  mock_imscope.table.return_value.__enter__.return_value = True
+  mock_imscope.tree_node_ex.return_value.__enter__.return_value = True
+  mock_imscope.child.return_value.__enter__.return_value = True
+  mock_imscope.window.return_value.__enter__.return_value = (True, True)
+  
   with patch.object(mock_app, '_populate_auto_slices') as mock_populate:
    mock_app._render_add_context_files_modal()
    
@@ -71,7 +78,8 @@ def test_add_all_triggers_auto_slices(mock_app):
  mock_app.files = [f_mock]
  mock_app.context_files = []
  
- with patch('src.gui_2.imgui') as mock_imgui:
+ with patch('src.gui_2.imgui') as mock_imgui, \
+      patch('src.gui_2.imscope') as mock_imscope:
   mock_imgui.button.side_effect = lambda label, size=None: label == "Add All##addall"
   mock_imgui.collapsing_header.return_value = True
   mock_imgui.begin_child.return_value = True
@@ -84,6 +92,12 @@ def test_add_all_triggers_auto_slices(mock_app):
   mock_imgui.TableColumnFlags_ = MagicMock()
   mock_imgui.TreeNodeFlags_ = MagicMock()
   
+  # Setup imscope mocks to act as context managers
+  mock_imscope.table.return_value.__enter__.return_value = True
+  mock_imscope.tree_node_ex.return_value.__enter__.return_value = True
+  mock_imscope.child.return_value.__enter__.return_value = True
+  mock_imscope.window.return_value.__enter__.return_value = (True, True)
+  
   with patch.object(mock_app, '_populate_auto_slices') as mock_populate:
    mock_app._render_context_composition_panel()