test formatting fixed

Author: uzairmukadam

tests/test_sheet_ql.py

@@ -20,7 +20,11 @@ def setUp(self):
 
         self.csv_path = os.path.join(self.test_dir, "sample.csv")
         pd.DataFrame(
-            {"ID": [1, 2, 3], "Name": ["Alice", "Bob", "Charlie"], "Value": [100, 200, 150]}
+            {
+                "ID": [1, 2, 3],
+                "Name": ["Alice", "Bob", "Charlie"],
+                "Value": [100, 200, 150],
+            }
         ).to_csv(self.csv_path, index=False)
 
         self.excel_path = os.path.join(self.test_dir, "sample.xlsx")
@@ -127,9 +131,9 @@ def test_06_yaml_script_interactive_run(self, mock_input):
     def test_07_load_command(self):
         """Test loading a new file mid-session with the '.load' command."""
         self.tool._register_dataframes(self.tool._load_data([self.csv_path]))
-        initial_tables = self.tool.db_connection.execute("SHOW TABLES;").fetchdf()[
-            "name"
-        ].tolist()
+        initial_tables = (
+            self.tool.db_connection.execute("SHOW TABLES;").fetchdf()["name"].tolist()
+        )
         self.assertIn("sample_csv", initial_tables)
         self.assertNotIn("sample_json", initial_tables)
 
@@ -138,9 +142,9 @@ def test_07_load_command(self):
         ):
             self.tool._add_new_files()
 
-        final_tables = self.tool.db_connection.execute("SHOW TABLES;").fetchdf()[
-            "name"
-        ].tolist()
+        final_tables = (
+            self.tool.db_connection.execute("SHOW TABLES;").fetchdf()["name"].tolist()
+        )
         self.assertIn("sample_csv", final_tables)
         self.assertIn("sample_json", final_tables)
 
@@ -168,7 +172,7 @@ def test_09_exit_command_with_save_prompt(self, mock_export):
         """Test that '.exit' prompts to save when results are staged."""
         self.tool.console.input.return_value = "y"
         self.tool.results_to_save["my_results"] = pd.DataFrame({"a": [1]})
-        
+
         should_exit = self.tool._handle_meta_command(".exit")
 
         self.assertTrue(should_exit)