scalebox-sdk 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

scalebox/test/test_code_interpreter_sync_comprehensive.py

@@ -106,7 +106,7 @@ result = {"x": x, "y": y}
 print(result)
 """
 
-        execution = self.sandbox.run_code(code,envs={"CI_TEST": "sync_test1"})
+        execution = self.sandbox.run_code(code, envs={"CI_TEST": "sync_test1"})
         print(execution.to_json())
         # time.sleep(1000)
         assert isinstance(execution, Execution)
@@ -366,10 +366,7 @@ print(f"结果: {result}")
         assert self.sandbox is not None
 
         # 创建Python上下文
-        python_context = self.sandbox.create_code_context(
-            language="python", 
-            cwd="/tmp"
-        )
+        python_context = self.sandbox.create_code_context(language="python", cwd="/tmp")
         print(python_context)
         assert isinstance(python_context, Context)
         assert python_context.id is not None

scalebox/test/test_connect_pause_async.py

@@ -78,7 +78,9 @@ class ConnectPauseAsyncTest:
             # 4. 创建目录和文件
             print("Step 4: Creating directory and nested files...")
             await self.sandbox.files.make_dir("/tmp/test_dir")
-            await self.sandbox.files.write("/tmp/test_dir/nested_file.txt", "Nested file content")
+            await self.sandbox.files.write(
+                "/tmp/test_dir/nested_file.txt", "Nested file content"
+            )
             assert await self.sandbox.files.exists("/tmp/test_dir")
             assert await self.sandbox.files.exists("/tmp/test_dir/nested_file.txt")
             print("Created directory and nested file")
@@ -106,13 +108,17 @@ class ConnectPauseAsyncTest:
             # 校验应该存在的文件
             exists = await connected_sandbox.files.exists("/tmp/test_file1.txt")
             assert exists, "test_file1.txt should exist"
-            file1_content = await connected_sandbox.files.read("/tmp/test_file1.txt", format="text")
+            file1_content = await connected_sandbox.files.read(
+                "/tmp/test_file1.txt", format="text"
+            )
             assert "Content of test file 1" in file1_content
             print("✅ Verified test_file1.txt exists and content is correct")
 
             exists = await connected_sandbox.files.exists("/tmp/test_file3.bin")
             assert exists, "test_file3.bin should exist"
-            file3_content = await connected_sandbox.files.read("/tmp/test_file3.bin", format="bytes")
+            file3_content = await connected_sandbox.files.read(
+                "/tmp/test_file3.bin", format="bytes"
+            )
             assert bytes(file3_content) == b"Binary content for test file 3"
             print("✅ Verified test_file3.bin exists and content is correct")
 
@@ -124,15 +130,21 @@ class ConnectPauseAsyncTest:
             # 校验目录和嵌套文件
             exists = await connected_sandbox.files.exists("/tmp/test_dir")
             assert exists, "test_dir should exist"
-            exists = await connected_sandbox.files.exists("/tmp/test_dir/nested_file.txt")
+            exists = await connected_sandbox.files.exists(
+                "/tmp/test_dir/nested_file.txt"
+            )
             assert exists, "nested_file.txt should exist"
-            nested_content = await connected_sandbox.files.read("/tmp/test_dir/nested_file.txt", format="text")
+            nested_content = await connected_sandbox.files.read(
+                "/tmp/test_dir/nested_file.txt", format="text"
+            )
             assert "Nested file content" in nested_content
             print("✅ Verified directory and nested file exist and content is correct")
 
             # 8. 在connect后的沙箱中执行一些操作验证功能正常
             print("Step 8: Verifying sandbox functionality after connect...")
-            result = await connected_sandbox.commands.run("echo 'Test command after connect'")
+            result = await connected_sandbox.commands.run(
+                "echo 'Test command after connect'"
+            )
             assert result.exit_code == 0
             assert "Test command after connect" in result.stdout
             print("✅ Verified sandbox commands work after connect")
@@ -140,20 +152,27 @@ class ConnectPauseAsyncTest:
             # 9. 创建新文件验证写入功能正常
             print("Step 9: Testing file write after connect...")
             new_file_path = "/tmp/new_file_after_connect.txt"
-            await connected_sandbox.files.write(new_file_path, "New file created after connect")
+            await connected_sandbox.files.write(
+                new_file_path, "New file created after connect"
+            )
             exists = await connected_sandbox.files.exists(new_file_path)
             assert exists
-            new_content = await connected_sandbox.files.read(new_file_path, format="text")
+            new_content = await connected_sandbox.files.read(
+                new_file_path, format="text"
+            )
             assert "New file created after connect" in new_content
             print("✅ Verified file write works after connect")
 
-            self.log_test_result("Connect and Pause with Files", True, "All verifications passed")
+            self.log_test_result(
+                "Connect and Pause with Files", True, "All verifications passed"
+            )
             return True
 
         except Exception as e:
             self.log_test_result("Connect and Pause with Files", False, str(e))
             print(f"Test failed with error: {e}")
             import traceback
+
             traceback.print_exc()
             return False
 
@@ -217,13 +236,18 @@ class ConnectPauseAsyncTest:
             # 清理
             await sandbox.kill()
 
-            self.log_test_result("Multiple Connect and Pause Operations", True, "All verifications passed")
+            self.log_test_result(
+                "Multiple Connect and Pause Operations",
+                True,
+                "All verifications passed",
+            )
             return True
 
         except Exception as e:
             self.log_test_result("Multiple Connect and Pause Operations", False, str(e))
             print(f"Test failed with error: {e}")
             import traceback
+
             traceback.print_exc()
             return False
 
@@ -274,4 +298,3 @@ async def main():
 
 if __name__ == "__main__":
     asyncio.run(main())
-

scalebox/test/test_connect_pause_sync.py

@@ -76,7 +76,9 @@ class ConnectPauseSyncTest:
             # 4. 创建目录和文件
             print("Step 4: Creating directory and nested files...")
             self.sandbox.files.make_dir("/tmp/test_dir")
-            self.sandbox.files.write("/tmp/test_dir/nested_file.txt", "Nested file content")
+            self.sandbox.files.write(
+                "/tmp/test_dir/nested_file.txt", "Nested file content"
+            )
             assert self.sandbox.files.exists("/tmp/test_dir")
             assert self.sandbox.files.exists("/tmp/test_dir/nested_file.txt")
             print("Created directory and nested file")
@@ -103,24 +105,40 @@ class ConnectPauseSyncTest:
             print("Step 7: Verifying file operations after connect...")
 
             # 校验应该存在的文件
-            assert connected_sandbox.files.exists("/tmp/test_file1.txt"), "test_file1.txt should exist"
-            file1_content = connected_sandbox.files.read("/tmp/test_file1.txt", format="text")
+            assert connected_sandbox.files.exists(
+                "/tmp/test_file1.txt"
+            ), "test_file1.txt should exist"
+            file1_content = connected_sandbox.files.read(
+                "/tmp/test_file1.txt", format="text"
+            )
             assert "Content of test file 1" in file1_content
             print("✅ Verified test_file1.txt exists and content is correct")
 
-            assert connected_sandbox.files.exists("/tmp/test_file3.bin"), "test_file3.bin should exist"
-            file3_content = connected_sandbox.files.read("/tmp/test_file3.bin", format="bytes")
+            assert connected_sandbox.files.exists(
+                "/tmp/test_file3.bin"
+            ), "test_file3.bin should exist"
+            file3_content = connected_sandbox.files.read(
+                "/tmp/test_file3.bin", format="bytes"
+            )
             assert bytes(file3_content) == b"Binary content for test file 3"
             print("✅ Verified test_file3.bin exists and content is correct")
 
             # 校验应该不存在的文件（被删除的）
-            assert not connected_sandbox.files.exists("/tmp/test_file2.txt"), "test_file2.txt should not exist"
+            assert not connected_sandbox.files.exists(
+                "/tmp/test_file2.txt"
+            ), "test_file2.txt should not exist"
             print("✅ Verified test_file2.txt was deleted correctly")
 
             # 校验目录和嵌套文件
-            assert connected_sandbox.files.exists("/tmp/test_dir"), "test_dir should exist"
-            assert connected_sandbox.files.exists("/tmp/test_dir/nested_file.txt"), "nested_file.txt should exist"
-            nested_content = connected_sandbox.files.read("/tmp/test_dir/nested_file.txt", format="text")
+            assert connected_sandbox.files.exists(
+                "/tmp/test_dir"
+            ), "test_dir should exist"
+            assert connected_sandbox.files.exists(
+                "/tmp/test_dir/nested_file.txt"
+            ), "nested_file.txt should exist"
+            nested_content = connected_sandbox.files.read(
+                "/tmp/test_dir/nested_file.txt", format="text"
+            )
             assert "Nested file content" in nested_content
             print("✅ Verified directory and nested file exist and content is correct")
 
@@ -134,19 +152,24 @@ class ConnectPauseSyncTest:
             # 9. 创建新文件验证写入功能正常
             print("Step 9: Testing file write after connect...")
             new_file_path = "/tmp/new_file_after_connect.txt"
-            connected_sandbox.files.write(new_file_path, "New file created after connect")
+            connected_sandbox.files.write(
+                new_file_path, "New file created after connect"
+            )
             assert connected_sandbox.files.exists(new_file_path)
             new_content = connected_sandbox.files.read(new_file_path, format="text")
             assert "New file created after connect" in new_content
             print("✅ Verified file write works after connect")
 
-            self.log_test_result("Connect and Pause with Files", True, "All verifications passed")
+            self.log_test_result(
+                "Connect and Pause with Files", True, "All verifications passed"
+            )
             return True
 
         except Exception as e:
             self.log_test_result("Connect and Pause with Files", False, str(e))
             print(f"Test failed with error: {e}")
             import traceback
+
             traceback.print_exc()
             return False
 
@@ -198,22 +221,33 @@ class ConnectPauseSyncTest:
             time.sleep(2)
 
             # 9. 验证第二次操作的结果
-            assert not sandbox.files.exists("/tmp/multi_test1.txt"), "multi_test1.txt should be deleted"
-            assert sandbox.files.exists("/tmp/multi_test2.txt"), "multi_test2.txt should exist"
-            assert sandbox.files.exists("/tmp/multi_test3.txt"), "multi_test3.txt should exist"
+            assert not sandbox.files.exists(
+                "/tmp/multi_test1.txt"
+            ), "multi_test1.txt should be deleted"
+            assert sandbox.files.exists(
+                "/tmp/multi_test2.txt"
+            ), "multi_test2.txt should exist"
+            assert sandbox.files.exists(
+                "/tmp/multi_test3.txt"
+            ), "multi_test3.txt should exist"
             content = sandbox.files.read("/tmp/multi_test3.txt", format="text")
             assert "Second batch" in content
 
             # 清理
             sandbox.kill()
 
-            self.log_test_result("Multiple Connect and Pause Operations", True, "All verifications passed")
+            self.log_test_result(
+                "Multiple Connect and Pause Operations",
+                True,
+                "All verifications passed",
+            )
             return True
 
         except Exception as e:
             self.log_test_result("Multiple Connect and Pause Operations", False, str(e))
             print(f"Test failed with error: {e}")
             import traceback
+
             traceback.print_exc()
             return False
 
@@ -264,4 +298,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-

scalebox/test/test_csx_desktop_examples.py

@@ -18,7 +18,9 @@ class CsxDesktopValidator:
         self.success_count = 0
         self.fail_count = 0
 
-    def log_test_result(self, name: str, success: bool, error: Optional[Exception] = None) -> None:
+    def log_test_result(
+        self, name: str, success: bool, error: Optional[Exception] = None
+    ) -> None:
         if success:
             self.success_count += 1
             print(f"✅ {name}")
@@ -126,5 +128,3 @@ if __name__ == "__main__":
     finally:
         validator.cleanup()
         validator.print_summary()
-
-

scalebox/test/test_desktop_sandbox_sf.py

@@ -17,7 +17,7 @@ def main():
     print("=" * 60)
     print("开始测试：基于desktop创建sandbox并执行test_sf.py")
     print("=" * 60)
-    
+
     # 1. 基于desktop创建sandbox
     print("\n[步骤 1] 正在创建desktop sandbox...")
     sandbox = Sandbox.create(
@@ -25,24 +25,24 @@ def main():
         timeout=3600,  # 1小时超时
     )
     print(f"✓ Sandbox创建成功，ID: {sandbox.sandbox_id}")
-    
+
     # 2. 获取network_proxy
     print("\n[步骤 2] 获取network_proxy...")
     network_proxy = sandbox.network_proxy()
     print(f"✓ Network Proxy: {network_proxy}")
-    
+
     # 3. 获取VNC URL（不启动，直接获取）
     print("\n[步骤 3] 获取VNC URL...")
     vnc_url = sandbox.stream.get_url(auto_connect=True)
     print(f"✓ VNC URL: {vnc_url}")
-    
+
     # 4. 读取test_sf.py并修改username和password
     print("\n[步骤 4] 读取并修改test_sf.py...")
     test_sf_path = os.path.join(os.path.dirname(__file__), "test_sf.py")
-    
-    with open(test_sf_path, 'r', encoding='utf-8') as f:
+
+    with open(test_sf_path, "r", encoding="utf-8") as f:
         test_sf_content = f.read()
-    
+
     # 从network_proxy中提取username和password
     if network_proxy and isinstance(network_proxy, dict):
         # network_proxy结构: {'proxy_configs': {'username': '...', 'password': '...', ...}, ...}
@@ -54,7 +54,7 @@ def main():
             # 兼容旧格式：直接在network_proxy中
             proxy_username = network_proxy.get("username", "")
             proxy_password = network_proxy.get("password", "")
-        
+
         if not proxy_username or not proxy_password:
             print("⚠ 警告: network_proxy中缺少username或password字段")
             print(f"  network_proxy内容: {network_proxy}")
@@ -63,36 +63,32 @@ def main():
             # 匹配proxy字典中的username和password（处理可能的引号差异）
             pattern_username = r'"username"\s*:\s*"[^"]*"'
             pattern_password = r'"password"\s*:\s*"[^"]*"'
-            
+
             test_sf_content = re.sub(
-                pattern_username,
-                f'"username": "{proxy_username}"',
-                test_sf_content
+                pattern_username, f'"username": "{proxy_username}"', test_sf_content
             )
             test_sf_content = re.sub(
-                pattern_password,
-                f'"password": "{proxy_password}"',
-                test_sf_content
+                pattern_password, f'"password": "{proxy_password}"', test_sf_content
             )
-            
+
             print(f"✓ 已更新username: {proxy_username}")
             print(f"✓ 已更新password: {proxy_password[:20]}...")
     else:
         print("⚠ 警告: network_proxy为空或格式不正确，跳过修改")
         print(f"  network_proxy类型: {type(network_proxy)}, 值: {network_proxy}")
-    
+
     # 5. 上传test_sf.py到sandbox的/目录
     print("\n[步骤 5] 上传test_sf.py到sandbox的/目录...")
-    result = sandbox.files.write("/test_sf.py", test_sf_content.encode('utf-8'))
+    result = sandbox.files.write("/test_sf.py", test_sf_content.encode("utf-8"))
     print(f"✓ 文件上传成功: {result.path}")
-    
+
     # 6. 在sandbox中执行命令
     print("\n[步骤 6] 在sandbox中执行: source /venv/bin/activate && python /test_sf.py")
     try:
         # 使用bash -c确保source命令和环境变量正确加载
         result = sandbox.commands.run(
             'bash -c "source /venv/bin/activate && python /test_sf.py"',
-            timeout=600  # 10分钟超时
+            timeout=600,  # 10分钟超时
         )
         print(f"✓ 命令执行完成")
         print(f"退出码: {result.exit_code}")
@@ -102,11 +98,11 @@ def main():
     except Exception as e:
         print(f"✗ 命令执行失败: {e}")
         raise
-    
+
     print("\n" + "=" * 60)
     print("测试完成！")
     print("=" * 60)
-    
+
     # 清理（可选，注释掉以保留sandbox）
     # sandbox.kill()
     # print("Sandbox已清理")
@@ -114,4 +110,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-

scalebox/test/test_download_url.py

@@ -6,17 +6,13 @@ print("正在准备文件内容...")
 results_data = {
     "experiment_id": "exp_001",
     "status": "completed",
-    "metrics": {
-        "accuracy": 0.952,
-        "precision": 0.94,
-        "recall": 0.96
-    },
+    "metrics": {"accuracy": 0.952, "precision": 0.94, "recall": 0.96},
     "timestamp": "2024-01-01T12:00:00Z",
     "details": [
         {"epoch": 1, "loss": 0.45},
         {"epoch": 2, "loss": 0.32},
-        {"epoch": 3, "loss": 0.18}
-    ]
+        {"epoch": 3, "loss": 0.18},
+    ],
 }
 
 # 转换为 JSON 字符串
@@ -34,16 +30,12 @@ print(f"✅ 文件已写入沙箱: {remote_path}")
 # 3. 获取预签名下载 URL
 print("\n正在生成下载链接...")
 download_url = sandbox.download_url(
-    path=remote_path,
-    use_signature_expiration=180  # 180秒有效期
+    path=remote_path, use_signature_expiration=180  # 180秒有效期
 )
 
 print(f"\n📥 下载 URL: {download_url}")
 print(f"⏰ 链接将在 3 分钟后过期")
 
-upload_url = sandbox.upload_url(
-    path="/home/user",
-    use_signature_expiration=360
-)
+upload_url = sandbox.upload_url(path="/home/user", use_signature_expiration=360)
 print(f"\n📥 上传 URL: {upload_url}")
-print(f"⏰ 链接将在 6 分钟后过期")
+print(f"⏰ 链接将在 6 分钟后过期")