unit test for mdagent learn flag

mdagent/tools/maketools.py

@@ -92,12 +92,12 @@ def make_all_tools(
         ModifyBaseSimulationScriptTool(path_registry=path_instance, llm=llm),
         SimulationOutputFigures(),
     ]
+    if subagent_settings is None:
+        subagent_settings = SubAgentSettings(path_registry=path_instance)
 
     # tools using subagents
     subagents_tools = []
     if not skip_subagents:
-        if subagent_settings is None:
-            subagent_settings = SubAgentSettings(path_registry=path_instance)
         subagents_tools = [
             CreateNewTool(subagent_settings=subagent_settings),
             RetryExecuteSkill(subagent_settings=subagent_settings),

tests/test_agent.py

@@ -3,6 +3,8 @@
 
 import pytest
 
+from mdagent.mainagent.agent import MDAgent
+from mdagent.mainagent.prompt import openaifxn_prompt, structured_prompt
 from mdagent.subagents.agents.action import Action
 from mdagent.subagents.agents.skill import SkillManager
 from mdagent.subagents.subagent_fxns import Iterator
@@ -231,3 +233,17 @@ def test_update_skill_library(skill_manager):
             path="/mock_dir/code/test_function.py",
             description="Code for new tool test_function",
         )
+
+
+# test initiating mdagent with learn = false
+def test_mdagent_init():
+    mdagent_skill = MDAgent(learn=False)
+    assert (
+        mdagent_skill.prompt == openaifxn_prompt
+        or mdagent_skill.prompt == structured_prompt
+    )
+    assert mdagent_skill.skip_subagents is True
+
+    mdagent_learn = MDAgent(learn=True)
+    assert mdagent_learn.prompt == openaifxn_prompt
+    assert mdagent_learn.skip_subagents is False