diff --git a/tests/unit/test_spinner.py b/tests/unit/test_spinner.py index 1e6d3b6e..ec75a687 100644 --- a/tests/unit/test_spinner.py +++ b/tests/unit/test_spinner.py @@ -1,11 +1,9 @@ - # Generated by CodiumAI import time import unittest + from autogpt.spinner import Spinner - - """ Code Analysis @@ -30,8 +28,8 @@ Fields: ALMOST_DONE_MESSAGE = "Almost done..." PLEASE_WAIT = "Please wait..." -class TestSpinner(unittest.TestCase): +class TestSpinner(unittest.TestCase): # Tests that the spinner initializes with default values. def test_spinner_initializes_with_default_values(self): with Spinner() as spinner: @@ -63,6 +61,7 @@ class TestSpinner(unittest.TestCase): self.assertFalse(spinner.running) # Tests that the spinner can be used as a context manager. + def test_spinner_can_be_used_as_context_manager(self): with Spinner() as spinner: self.assertTrue(spinner.running)