Add test for reconnect

Author: jakubno

README.md

@@ -89,7 +89,7 @@ with CodeInterpreter() as sandbox:
     result = sandbox.notebook.exec_cell(code)
 
 # there's your image
-image = result.display_data[0]["image/png"]
+image = result.data[0].png
 
 # example how to show the image / prove it works
 i = base64.b64decode(image)

js/tests/reconnect.test.ts

@@ -0,0 +1,16 @@
+import { CodeInterpreter } from '../src'
+
+import { expect, test } from 'vitest'
+
+test('reconnect', async () => {
+  let sandbox = await CodeInterpreter.create()
+  await sandbox.close()
+
+  sandbox = await CodeInterpreter.reconnect(sandbox.id)
+
+  const result = await sandbox.notebook.execCell('x =1; x')
+
+  expect(result.text).toEqual('1')
+
+  await sandbox.close()
+})

python/tests/test_reconnect.py

@@ -0,0 +1,12 @@
+from e2b_code_interpreter.main import CodeInterpreter
+
+
+def test_basic():
+    with CodeInterpreter() as sandbox:
+        sandbox_id = sandbox.id
+
+    sandbox = CodeInterpreter.reconnect(sandbox_id)
+    result = sandbox.notebook.exec_cell("x =1; x")
+    assert result.text == "1"
+
+    sandbox.close()