Test Cases (#2993)

### What problem does this PR solve?

Test Cases

### Type of change

- [x] Refactoring

Co-authored-by: liuhua <10215101452@stu.ecun.edu.cn>

sdk/python/test/t_dataset.py

@@ -1,53 +1,54 @@
-from ragflow import RAGFlow, DataSet
+from ragflow import RAGFlow
+import random
+import pytest
 
-from common import API_KEY, HOST_ADDRESS
-from test_sdkbase import TestSdk
+HOST_ADDRESS = 'http://127.0.0.1:9380'
+
+def test_create_dataset_with_name(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    rag.create_dataset("test_create_dataset_with_name")
+
+def test_create_dataset_with_duplicated_name(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    with pytest.raises(Exception) as exc_info:
+        rag.create_dataset("test_create_dataset_with_name")
+    assert str(exc_info.value) == "Duplicated dataset name in creating dataset."
+
+def test_create_dataset_with_random_chunk_method(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    valid_chunk_methods = ["naive","manual","qa","table","paper","book","laws","presentation","picture","one","knowledge_graph","email"]
+    random_chunk_method = random.choice(valid_chunk_methods)
+    rag.create_dataset("test_create_dataset_with_random_chunk_method",chunk_method=random_chunk_method)
+
+def test_create_dataset_with_invalid_parameter(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    valid_chunk_methods = ["naive", "manual", "qa", "table", "paper", "book", "laws", "presentation", "picture", "one",
+                           "knowledge_graph", "email"]
+    chunk_method = "invalid_chunk_method"
+    with pytest.raises(Exception) as exc_info:
+        rag.create_dataset("test_create_dataset_with_name",chunk_method=chunk_method)
+    assert str(exc_info.value) == f"{chunk_method} is not in {valid_chunk_methods}"
 
 
-class TestDataset(TestSdk):
-    def test_create_dataset_with_success(self):
-        """
-        Test creating a dataset with success
-        """
-        rag = RAGFlow(API_KEY, HOST_ADDRESS)
-        ds = rag.create_dataset("God")
-        if isinstance(ds, DataSet):
-            assert ds.name == "God", "Name does not match."
-        else:
-            assert False, f"Failed to create dataset, error: {ds}"
+def test_update_dataset_with_name(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    ds = rag.create_dataset("test_update_dataset")
+    ds.update({"name": "updated_dataset"})
 
-    def test_update_dataset_with_success(self):
-        """
-        Test updating a dataset with success.
-        """
-        rag = RAGFlow(API_KEY, HOST_ADDRESS)
-        ds = rag.create_dataset("ABC")
-        if isinstance(ds, DataSet):
-            assert ds.name == "ABC", "Name does not match."
-            res = ds.update({"name":"DEF"})
-            assert res is None, f"Failed to update dataset, error: {res}"
-        else:
-            assert False, f"Failed to create dataset, error: {ds}"
 
-    def test_delete_datasets_with_success(self):
-        """
-        Test deleting a dataset with success
-        """
-        rag = RAGFlow(API_KEY, HOST_ADDRESS)
-        ds = rag.create_dataset("MA")
-        if isinstance(ds, DataSet):
-            assert ds.name == "MA", "Name does not match."
-            res = rag.delete_datasets(ids=[ds.id])
-            assert res is None, f"Failed to delete dataset, error: {res}"
-        else:
-            assert False, f"Failed to create dataset, error: {ds}"
+def test_delete_datasets_with_success(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    ds = rag.create_dataset("MA")
+    rag.delete_datasets(ids=[ds.id])
 
-    def test_list_datasets_with_success(self):
-        """
-        Test listing datasets with success
-        """
-        rag = RAGFlow(API_KEY, HOST_ADDRESS)
-        list_datasets = rag.list_datasets()
-        assert len(list_datasets) > 0, "Do not exist any dataset"
-        for ds in list_datasets:
-            assert isinstance(ds, DataSet), "Existence type is not dataset."
+
+def test_list_datasets_with_success(get_api_key_fixture):
+    API_KEY = get_api_key_fixture
+    rag = RAGFlow(API_KEY, HOST_ADDRESS)
+    rag.list_datasets()