fix: file_handler issue (#869)

* fix: file_handler issue

* fix: add logic for the trained_agent data
This commit is contained in:
Eduardo Chiarotti
2024-07-04 16:34:43 -03:00
committed by GitHub
parent 4bcd1df6bb
commit 81ed6f177e
3 changed files with 13 additions and 7 deletions

View File

@@ -17,6 +17,10 @@ class TestPickleHandler(unittest.TestCase):
os.remove(self.file_path)
def test_initialize_file(self):
assert os.path.exists(self.file_path) is False
self.handler.initialize_file()
assert os.path.exists(self.file_path) is True
assert os.path.getsize(self.file_path) >= 0