diff --git a/test/utils/test_kaggle.py b/test/utils/test_kaggle.py index bd6e2693..cadb7dd1 100644 --- a/test/utils/test_kaggle.py +++ b/test/utils/test_kaggle.py @@ -13,18 +13,21 @@ from rdagent.utils.agent.tpl import T class TestTpl(unittest.TestCase): def test_competition_template(self): + """ + export KG_COMPETITION= before running this test + """ competition = KAGGLE_IMPLEMENT_SETTING.competition print(f"[bold orange]{competition}[/bold orange]") ws = KGFBWorkspace( template_folder_path=Path(__file__).parent.parent.parent / "rdagent/scenarios/kaggle/experiment" - / f"{KAGGLE_IMPLEMENT_SETTING.competition}_template" + / f"{competition}_template" ) print(ws.workspace_path) ws.execute() success = (ws.workspace_path / "submission.csv").exists() + self.assertTrue(success, "submission.csv is not generated") ws.clear() - return success if __name__ == "__main__":