update test

Author: Lingling Peng

tests/integration/synapseclient/models/async/test_grid_async.py

@@ -188,7 +188,6 @@ async def test_import_csv_to_grid_session_async(
         self,
         project_model,
         record_set_fixture: RecordSet,
-        schedule_for_cleanup: Callable[..., None],
     ) -> None:
         """Test importing a CSV file into a grid session."""
 
@@ -220,7 +219,7 @@ async def test_import_csv_to_grid_session_async(
                 path=temp_csv.name, parent_id=project_model.id
             ).store_async(synapse_client=self.syn)
 
-            schedule_for_cleanup(file.id)
+            self.schedule_for_cleanup(file.id)
 
         # WHEN: Importing the CSV into the grid session
         imported_grid = await created_grid.import_csv_async(

tests/unit/synapseclient/models/async/unit_test_curation_async.py

@@ -862,15 +862,11 @@ async def test_import_csv_async(self) -> None:
             updated_count=1,
         )
 
-        # Set up mock preview class: constructor returns a mock whose
-        # send_job_and_wait_async returns the mock_preview_response
         mock_preview_instance = MagicMock()
         mock_preview_instance.send_job_and_wait_async = AsyncMock(
             return_value=mock_preview_response
         )
 
-        # Set up mock import class: constructor returns a mock whose
-        # send_job_and_wait_async returns the mock_import_response
         mock_import_instance = MagicMock()
         mock_import_instance.send_job_and_wait_async = AsyncMock(
             return_value=mock_import_response