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 too expensive to run on CI")
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 draw_image_agent, run_agent_step_by_step
 
 
 
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 = run_agent_step_by_step(prompt, 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()