Spaces:
Runtime error
Runtime error
Charles Azam commited on
Commit ·
1ea134d
1
Parent(s): 1eb9c9d
feat: remove skipping tests
Browse files
tests/deepsearch/test_pdf_agent.py
CHANGED
|
@@ -29,7 +29,6 @@ def test_pdf_agent():
|
|
| 29 |
FindInMarkdownTool(ocr_response).forward(["thermal neutron", "neutron"])
|
| 30 |
|
| 31 |
|
| 32 |
-
@pytest.mark.skip(reason="This test is too expensive to run on CI")
|
| 33 |
def test_run_pdf_agent():
|
| 34 |
ocr_response = load_mock_ocr_response()
|
| 35 |
pdf_agent = create_agent(ocr_response)
|
|
|
|
| 29 |
FindInMarkdownTool(ocr_response).forward(["thermal neutron", "neutron"])
|
| 30 |
|
| 31 |
|
|
|
|
| 32 |
def test_run_pdf_agent():
|
| 33 |
ocr_response = load_mock_ocr_response()
|
| 34 |
pdf_agent = create_agent(ocr_response)
|
tests/deepsearch/test_web_agent.py
CHANGED
|
@@ -6,7 +6,7 @@ def test_create_web_search_agent():
|
|
| 6 |
create_web_search_agent()
|
| 7 |
|
| 8 |
|
| 9 |
-
@pytest.mark.skip(reason="This test is
|
| 10 |
def test_run_web_search_agent():
|
| 11 |
agent = create_web_search_agent()
|
| 12 |
assert (
|
|
|
|
| 6 |
create_web_search_agent()
|
| 7 |
|
| 8 |
|
| 9 |
+
@pytest.mark.skip(reason="This test is very long to run")
|
| 10 |
def test_run_web_search_agent():
|
| 11 |
agent = create_web_search_agent()
|
| 12 |
assert (
|
tests/webcrawler/test_draw_agent.py
CHANGED
|
@@ -1,5 +1,8 @@
|
|
| 1 |
import pytest
|
| 2 |
-
from deepengineer.deepsearch.draw_agent import
|
|
|
|
|
|
|
|
|
|
| 3 |
from deepengineer.common_path import DATA_DIR
|
| 4 |
from pathlib import Path
|
| 5 |
|
|
@@ -18,5 +21,5 @@ def test_run_agent_step_by_step():
|
|
| 18 |
prompt = """Propose moi un schéma très détaillé d'un réacteur nucléaire hélium graphite."""
|
| 19 |
output_path = Path(DATA_DIR) / "figure.png"
|
| 20 |
output_path.unlink(missing_ok=True)
|
| 21 |
-
output_path =
|
| 22 |
assert output_path.exists()
|
|
|
|
| 1 |
import pytest
|
| 2 |
+
from deepengineer.deepsearch.draw_agent import (
|
| 3 |
+
draw_image_agent,
|
| 4 |
+
multiple_steps_draw_image_agent,
|
| 5 |
+
)
|
| 6 |
from deepengineer.common_path import DATA_DIR
|
| 7 |
from pathlib import Path
|
| 8 |
|
|
|
|
| 21 |
prompt = """Propose moi un schéma très détaillé d'un réacteur nucléaire hélium graphite."""
|
| 22 |
output_path = Path(DATA_DIR) / "figure.png"
|
| 23 |
output_path.unlink(missing_ok=True)
|
| 24 |
+
output_path = multiple_steps_draw_image_agent(prompt, output_path)
|
| 25 |
assert output_path.exists()
|