'Refactored by Sourcery' (#736)

Co-authored-by: Sourcery AI <>

Author: sourcery-ai[bot]

tests/test_codemanager.py

@@ -73,8 +73,7 @@ def code_manager(self, smart_dataframe: SmartDataframe):
 
     @pytest.fixture
     def exec_context(self) -> MagicMock:
-        context = MagicMock(spec=CodeExecutionContext)
-        return context
+        return MagicMock(spec=CodeExecutionContext)
 
     def test_run_code_for_calculations(
         self, code_manager: CodeManager, exec_context: MagicMock