add basic snapshot testing to python-sdk

packages/python-sdk/tests/async/sandbox_async/test_snapshot.py

@@ -0,0 +1,17 @@
+import pytest
+from e2b import AsyncSandbox
+
+
+@pytest.mark.skip_debug()
+async def test_snapshot(template):
+    sbx = await AsyncSandbox.create(template, timeout=5)
+    try:
+        assert await sbx.is_running()
+
+        sandbox_id = await sbx.pause()
+        assert not await sbx.is_running()
+
+        await sbx.resume(sandbox_id)
+        assert await sbx.is_running()
+    finally:
+        await sbx.kill()

packages/python-sdk/tests/conftest.py

@@ -1,22 +1,23 @@
-import pytest
-import pytest_asyncio
 import os
-
 from logging import warning
 
-from e2b import Sandbox, AsyncSandbox
+import pytest
+import pytest_asyncio
+from e2b import AsyncSandbox, Sandbox
 
 
 @pytest.fixture()
 def template():
-    return "base"
+    return "base_01"
 
 
 @pytest.fixture()
 def sandbox(template, debug):
     sandbox = Sandbox(template)
 
     try:
+        sandbox_id = sandbox.pause()
+        sandbox.resume(sandbox_id)
         yield sandbox
     finally:
         try:
@@ -33,6 +34,8 @@ async def async_sandbox(template, debug):
     sandbox = await AsyncSandbox.create(template)
 
     try:
+        sandbox_id = await sandbox.pause()
+        await sandbox.resume(sandbox_id)
         yield sandbox
     finally:
         try:

packages/python-sdk/tests/sync/sandbox_sync/test_snapshot.py

@@ -0,0 +1,17 @@
+import pytest
+from e2b import Sandbox
+
+
+@pytest.mark.skip_debug()
+def test_snapshot(template):
+    sbx = Sandbox(template, timeout=5)
+    try:
+        assert sbx.is_running()
+
+        sandbox_id = sbx.pause()
+        assert not sbx.is_running()
+
+        sbx.resume(sandbox_id)
+        assert sbx.is_running()
+    finally:
+        sbx.kill()