mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-10-21 05:59:54 +08:00
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
import os
|
|
import asyncio
|
|
import unittest
|
|
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from app.tools.e2b_interpreter import E2BCodeInterpreter
|
|
from app.utils.common_utils import create_work_dir
|
|
|
|
try:
|
|
from dotenv import load_dotenv
|
|
except ModuleNotFoundError: # Fallback if python-dotenv is not installed
|
|
def load_dotenv(*args, **kwargs):
|
|
return None
|
|
|
|
try:
|
|
from app.tools.e2b_interpreter import E2BCodeInterpreter
|
|
except ModuleNotFoundError:
|
|
E2BCodeInterpreter = None
|
|
from app.utils.common_utils import create_task_id, create_work_dir
|
|
|
|
|
|
|
|
class TestE2BCodeInterpreter(unittest.TestCase):
|
|
def setUp(self):
|
|
load_dotenv()
|
|
|
|
if E2BCodeInterpreter is None:
|
|
self.skipTest("e2b_code_interpreter not available")
|
|
_, dirs = create_work_dir("20250312-104132-d3625cab")
|
|
notebook = NotebookSerializer(dirs["jupyter"])
|
|
|
|
self.code_interpreter = E2BCodeInterpreter(
|
|
self.task_id, self.work_dir, notebook
|
|
)
|
|
|
|
def test_execute_code(self):
|
|
if not os.getenv("E2B_API_KEY"):
|
|
self.skipTest("E2B_API_KEY not set")
|
|
|
|
code = """
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
# 生成数据
|
|
x = np.linspace(0, 2 * np.pi, 100) # x从0到2π，生成100个点
|
|
y = np.sin(x) # 计算对应的sin(x)值
|
|
|
|
# 绘图
|
|
plt.figure(figsize=(8, 4)) # 设置画布大小
|
|
plt.plot(x, y, label='y = sin(x)') # 绘制曲线，并添加图例
|
|
|
|
# 添加标签和标题
|
|
plt.title("Simple Sine Function")
|
|
plt.xlabel("x")
|
|
plt.ylabel("y")
|
|
|
|
# 添加网格和图例
|
|
plt.grid(True)
|
|
plt.legend()
|
|
|
|
# 显示图像
|
|
plt.show()
|
|
"""
|
|
asyncio.run(self.code_interpreter.initialize())
|
|
asyncio.run(self.code_interpreter.execute_code(code))
|
|
|