Apply automatic linting fixes to tests directory

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-12 13:31:07 +00:00
parent ad1ea46bbb
commit 46621113af
62 changed files with 1738 additions and 1821 deletions

View File

@@ -7,16 +7,16 @@ from crewai.utilities.file_handler import PickleHandler
class TestPickleHandler(unittest.TestCase):
def setUp(self):
def setUp(self) -> None:
self.file_name = "test_data.pkl"
self.file_path = os.path.join(os.getcwd(), self.file_name)
self.handler = PickleHandler(self.file_name)
def tearDown(self):
def tearDown(self) -> None:
if os.path.exists(self.file_path):
os.remove(self.file_path)
def test_initialize_file(self):
def test_initialize_file(self) -> None:
assert os.path.exists(self.file_path) is False
self.handler.initialize_file()
@@ -24,17 +24,17 @@ class TestPickleHandler(unittest.TestCase):
assert os.path.exists(self.file_path) is True
assert os.path.getsize(self.file_path) >= 0
def test_save_and_load(self):
def test_save_and_load(self) -> None:
data = {"key": "value"}
self.handler.save(data)
loaded_data = self.handler.load()
assert loaded_data == data
def test_load_empty_file(self):
def test_load_empty_file(self) -> None:
loaded_data = self.handler.load()
assert loaded_data == {}
def test_load_corrupted_file(self):
def test_load_corrupted_file(self) -> None:
with open(self.file_path, "wb") as file:
file.write(b"corrupted data")
@@ -42,4 +42,4 @@ class TestPickleHandler(unittest.TestCase):
self.handler.load()
assert str(exc.value) == "pickle data was truncated"
assert "<class '_pickle.UnpicklingError'>" == str(exc.type)
assert str(exc.type) == "<class '_pickle.UnpicklingError'>"