Fix unit test

Author: jhamon

tests/unit/data/test_bulk_import.py

@@ -6,7 +6,7 @@
     ImportErrorMode as ImportErrorModeGeneratedClass,
 )
 
-from pinecone.db_data.features.bulk_import import ImportFeatureMixin, ImportErrorMode
+from pinecone.db_data.resources.sync.bulk_import import BulkImportResource, ImportErrorMode
 
 
 def build_client_w_faked_response(mocker, body: str, status: int = 200):
@@ -19,19 +19,19 @@ def build_client_w_faked_response(mocker, body: str, status: int = 200):
     mock_request = mocker.patch.object(
         api_client.rest_client.pool_manager, "request", return_value=response
     )
-    return ImportFeatureMixin(api_client=api_client), mock_request
+    return BulkImportResource(api_client=api_client), mock_request
 
 
 class TestBulkImportStartImport:
-    def test_start_import_minimal(self, mocker):
+    def test_start_minimal(self, mocker):
         body = """
         {
             "id": "1"
         }
         """
         client, mock_req = build_client_w_faked_response(mocker, body)
 
-        my_import = client.start_import("s3://path/to/file.parquet")
+        my_import = client.start("s3://path/to/file.parquet")
 
         # We made some overrides to the print behavior, so we need to
         # call it to ensure it doesn't raise an exception
@@ -42,17 +42,15 @@ def test_start_import_minimal(self, mocker):
         assert my_import.to_dict() == {"id": "1"}
         assert my_import.__class__ == StartImportResponse
 
-    def test_start_import_with_kwargs(self, mocker):
+    def test_start_with_kwargs(self, mocker):
         body = """
         {
             "id": "1"
         }
         """
         client, mock_req = build_client_w_faked_response(mocker, body)
 
-        my_import = client.start_import(
-            uri="s3://path/to/file.parquet", integration_id="123-456-789"
-        )
+        my_import = client.start(uri="s3://path/to/file.parquet", integration_id="123-456-789")
         assert my_import.id == "1"
         assert my_import["id"] == "1"
         assert my_import.to_dict() == {"id": "1"}
@@ -68,37 +66,37 @@ def test_start_import_with_kwargs(self, mocker):
     @pytest.mark.parametrize(
         "error_mode_input", [ImportErrorMode.CONTINUE, "Continue", "continue", "cONTINUE"]
     )
-    def test_start_import_with_explicit_error_mode(self, mocker, error_mode_input):
+    def test_start_with_explicit_error_mode(self, mocker, error_mode_input):
         body = """
         {
             "id": "1"
         }
         """
         client, mock_req = build_client_w_faked_response(mocker, body)
 
-        client.start_import(uri="s3://path/to/file.parquet", error_mode=error_mode_input)
+        client.start(uri="s3://path/to/file.parquet", error_mode=error_mode_input)
         _, call_kwargs = mock_req.call_args
         assert (
             call_kwargs["body"]
             == '{"uri": "s3://path/to/file.parquet", "errorMode": {"onError": "continue"}}'
         )
 
-    def test_start_import_with_abort_error_mode(self, mocker):
+    def test_start_with_abort_error_mode(self, mocker):
         body = """
         {
             "id": "1"
         }
         """
         client, mock_req = build_client_w_faked_response(mocker, body)
 
-        client.start_import(uri="s3://path/to/file.parquet", error_mode=ImportErrorMode.ABORT)
+        client.start(uri="s3://path/to/file.parquet", error_mode=ImportErrorMode.ABORT)
         _, call_kwargs = mock_req.call_args
         assert (
             call_kwargs["body"]
             == '{"uri": "s3://path/to/file.parquet", "errorMode": {"onError": "abort"}}'
         )
 
-    def test_start_import_with_unknown_error_mode(self, mocker):
+    def test_start_with_unknown_error_mode(self, mocker):
         body = """
         {
             "id": "1"
@@ -107,7 +105,7 @@ def test_start_import_with_unknown_error_mode(self, mocker):
         client, mock_req = build_client_w_faked_response(mocker, body)
 
         with pytest.raises(ValueError) as e:
-            client.start_import(uri="s3://path/to/file.parquet", error_mode="unknown")
+            client.start(uri="s3://path/to/file.parquet", error_mode="unknown")
 
         assert "Invalid error_mode: unknown" in str(e.value)
 
@@ -122,7 +120,7 @@ def test_start_invalid_uri(self, mocker):
         client, mock_req = build_client_w_faked_response(mocker, body, 400)
 
         with pytest.raises(PineconeApiException) as e:
-            client.start_import(uri="invalid path")
+            client.start(uri="invalid path")
 
         assert e.value.status == 400
         assert e.value.body == body
@@ -134,7 +132,7 @@ def test_no_arguments(self, mocker):
         client, mock_req = build_client_w_faked_response(mocker, "")
 
         with pytest.raises(TypeError) as e:
-            client.start_import()
+            client.start()
 
         assert "missing 1 required positional argument" in str(e.value)
 
@@ -162,7 +160,7 @@ def test_describe_import(self, mocker):
         """
         client, mock_req = build_client_w_faked_response(mocker, body)
 
-        my_import = client.describe_import(id="1")
+        my_import = client.describe(id="1")
 
         # We made some overrides to the print behavior, so we need to
         # call it to ensure it doesn't raise an exception