#1629 - Added assertion in the test that checks the length of the all transactions file too.

- Added supporting function on the TempPrimaiteSession class that reads the all transactions csv file.
- Some renaming of the functions.
This commit is contained in:
Chris McCarthy
2023-07-17 12:14:47 +01:00
parent 75c91b9eb9
commit 360eb38c2b
5 changed files with 48 additions and 12 deletions

View File

@@ -19,4 +19,5 @@ def test_primaite_session(temp_primaite_session):
assert session_path.exists()
session.learn()
assert len(session.learn_av_reward_per_episode().keys()) == 10
assert len(session.learn_av_reward_per_episode_dict().keys()) == 10
assert len(session.learn_all_transactions_dict().keys()) == 10 * 256