From 82d7c168fe40f0e085ba9b9e737b3547cc882673 Mon Sep 17 00:00:00 2001 From: Chris McCarthy Date: Thu, 6 Jul 2023 14:13:02 +0100 Subject: [PATCH] #1593 - Check that agent saved file exists --- tests/test_primaite_session.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_primaite_session.py b/tests/test_primaite_session.py index ae0b0870..75ea5882 100644 --- a/tests/test_primaite_session.py +++ b/tests/test_primaite_session.py @@ -33,6 +33,9 @@ def test_primaite_session(temp_primaite_session): # Check that the network png file exists assert (session_path / f"network_{session.timestamp_str}.png").exists() + # Check that the saved agent exists + assert session._agent_session._saved_agent_path.exists() + # Check that both the transactions and av reward csv files exist for file in session.learning_path.iterdir(): if file.suffix == ".csv":