Spaces:
Sleeping
Sleeping
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()
|