Test E2B Executor

.github/workflows/tests.yml

@@ -78,6 +78,11 @@ jobs:
           uv run pytest ./tests/test_local_python_executor.py
         if: ${{ success() || failure() }}
 
+      - name: E2B executor tests
+        run: |
+          uv run pytest ./tests/test_e2b_executor.py
+        if: ${{ success() || failure() }}
+
       - name: Search tests
         run: |
           uv run pytest ./tests/test_search.py

tests/test_e2b_executor.py

@@ -0,0 +1,18 @@
+from unittest.mock import MagicMock, patch
+
+from smolagents.e2b_executor import E2BExecutor
+
+
+class TestE2BExecutor:
+    def test_e2b_executor_instantiation(self):
+        logger = MagicMock()
+        with patch("e2b_code_interpreter.Sandbox") as mock_sandbox:
+            mock_sandbox.return_value.commands.run.return_value.error = None
+            mock_sandbox.return_value.run_code.return_value.error = None
+            executor = E2BExecutor(additional_imports=[], tools=[], logger=logger)
+        assert isinstance(executor, E2BExecutor)
+        assert executor.logger == logger
+        assert executor.final_answer is False
+        assert executor.custom_tools == {}
+        assert executor.final_answer_pattern.pattern == r"final_answer\((.*?)\)"
+        assert executor.sbx == mock_sandbox.return_value