tests/test_one_shot_engineer.py

from pathlib import Path
import cmbagent



def test_one_shot_engineer():

   task = r"""
   Compute the sum of the first 1000 natural numbers.
   Plot the histogram of the numbers.
   """


   results = cmbagent.one_shot(
      task,
      max_rounds=20,
      agent="engineer",
      engineer_model="gpt-4.1",
      # engineer_model="gemini-2.5-flash",  # Now fixed with shared state
      # work_dir=str(tmp_path / "one_shot"), 
      work_dir="/Users/boris/Desktop/one_shot_engineer",
      clear_work_dir=True,
   )
   
   assert results is not None



test_one_shot_engineer()