superb-ai-onprem 0.5.9__py3-none-any.whl → 0.5.11__py3-none-any.whl

tests/data/test_data_service.py

@@ -0,0 +1,412 @@
+import pytest
+from unittest.mock import Mock, patch
+
+from spb_onprem.data.service import DataService
+from spb_onprem.data.queries import Queries
+from spb_onprem.data.entities import Data
+from spb_onprem.exceptions import BadParameterError
+
+
+class TestDataService:
+    """Test cases for DataService with new get_data_detail method."""
+
+    def setup_method(self):
+        """Set up test fixtures before each test method."""
+        self.data_service = DataService()
+        self.data_service.request_gql = Mock()
+
+    def test_get_data_detail_success(self):
+        """Test successful data detail retrieval."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        mock_response = {
+            "data": {
+                "id": data_id,
+                "datasetId": dataset_id,
+                "scene": [
+                    {
+                        "id": "scene-1",
+                        "content": {"id": "content-1"}
+                    }
+                ],
+                "annotation": {
+                    "versions": [
+                        {
+                            "id": "annotation-1",
+                            "content": {"id": "content-2"}
+                        }
+                    ]
+                },
+                "predictions": [
+                    {
+                        "id": "prediction-1",
+                        "content": {"id": "content-3"}
+                    }
+                ],
+                "thumbnail": {
+                    "id": "thumbnail-1"
+                }
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_data_detail(dataset_id, data_id)
+
+        # Assert
+        assert isinstance(result, Data)
+        assert result.id == data_id
+        assert result.dataset_id == dataset_id
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.GET_DETAIL,
+            Queries.GET_DETAIL["variables"](dataset_id, data_id)
+        )
+
+    def test_get_data_detail_missing_dataset_id(self):
+        """Test get_data_detail with missing dataset_id."""
+        # Arrange
+        dataset_id = None
+        data_id = "data-456"
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="dataset_id is required"):
+            self.data_service.get_data_detail(dataset_id, data_id)
+
+    def test_get_data_detail_missing_data_id(self):
+        """Test get_data_detail with missing data_id."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = None
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="data_id is required"):
+            self.data_service.get_data_detail(dataset_id, data_id)
+
+    def test_get_data_detail_empty_response(self):
+        """Test get_data_detail with empty data response."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        mock_response = {"data": {}}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_data_detail(dataset_id, data_id)
+
+        # Assert
+        assert isinstance(result, Data)
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.GET_DETAIL,
+            Queries.GET_DETAIL["variables"](dataset_id, data_id)
+        )
+
+    def test_get_data_detail_missing_data_field(self):
+        """Test get_data_detail with missing data field in response."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        mock_response = {}  # No data field
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_data_detail(dataset_id, data_id)
+
+        # Assert
+        assert isinstance(result, Data)
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.GET_DETAIL,
+            Queries.GET_DETAIL["variables"](dataset_id, data_id)
+        )
+
+    def test_get_data_detail_query_structure(self):
+        """Test that GET_DETAIL query has correct structure."""
+        # Arrange & Act
+        query_structure = Queries.GET_DETAIL
+
+        # Assert
+        assert query_structure["name"] == "getDataDetail"
+        assert "query GetDataDetail($datasetId: String!, $id: String!)" in query_structure["query"]
+        assert "data(datasetId: $datasetId, id: $id)" in query_structure["query"]
+        assert callable(query_structure["variables"])
+
+    def test_get_data_detail_variables_function(self):
+        """Test that variables function generates correct parameters."""
+        # Arrange
+        dataset_id = "dataset-789"
+        data_id = "data-101"
+
+        # Act
+        variables = Queries.GET_DETAIL["variables"](dataset_id, data_id)
+
+        # Assert
+        assert variables == {
+            "datasetId": dataset_id,
+            "id": data_id
+        }
+
+    def test_get_data_detail_with_nested_relationships(self):
+        """Test get_data_detail with complex nested relationships."""
+        # Arrange
+        dataset_id = "dataset-complex"
+        data_id = "data-complex"
+        mock_response = {
+            "data": {
+                "id": data_id,
+                "datasetId": dataset_id,
+                "scene": [
+                    {
+                        "id": "scene-1",
+                        "content": {"id": "scene-content-1"}
+                    },
+                    {
+                        "id": "scene-2", 
+                        "content": {"id": "scene-content-2"}
+                    }
+                ],
+                "annotation": {
+                    "versions": [
+                        {
+                            "id": "annotation-v1",
+                            "content": {"id": "annotation-content-1"}
+                        },
+                        {
+                            "id": "annotation-v2",
+                            "content": {"id": "annotation-content-2"}
+                        }
+                    ]
+                },
+                "predictions": [
+                    {
+                        "id": "prediction-1",
+                        "content": {"id": "prediction-content-1"}
+                    }
+                ],
+                "thumbnail": {
+                    "id": "thumbnail-complex"
+                }
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_data_detail(dataset_id, data_id)
+
+        # Assert
+        assert isinstance(result, Data)
+        assert result.id == data_id
+        assert len(result.scene) == 2
+        assert len(result.annotation.versions) == 2
+        assert len(result.predictions) == 1
+        assert result.thumbnail.id == "thumbnail-complex"
+
+    def test_get_data_detail_exception_handling(self):
+        """Test get_data_detail exception handling."""
+        # Arrange
+        dataset_id = "dataset-error"
+        data_id = "data-error"
+        self.data_service.request_gql.side_effect = Exception("GraphQL error")
+
+        # Act & Assert
+        with pytest.raises(Exception, match="GraphQL error"):
+            self.data_service.get_data_detail(dataset_id, data_id)
+
+        self.data_service.request_gql.assert_called_once()
+
+    def test_get_evaluation_value_list_success(self):
+        """Test successful evaluation value list retrieval."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        filter_dict = {"score": {"min": 0.8}}
+        length = 25
+        cursor = "cursor-abc"
+        
+        mock_response = {
+            "evaluationValueList": {
+                "totalCount": 100,
+                "next": "cursor-def",
+                "data": [
+                    {"dataId": "data-1"},
+                    {"dataId": "data-2"},
+                    {"dataId": "data-3"}
+                ]
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_evaluation_value_list(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id,
+            filter=filter_dict,
+            length=length,
+            cursor=cursor
+        )
+
+        # Assert
+        expected_result = {
+            "totalCount": 100,
+            "next": "cursor-def",
+            "data": [
+                {"dataId": "data-1"},
+                {"dataId": "data-2"},
+                {"dataId": "data-3"}
+            ]
+        }
+        assert result == expected_result
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.GET_EVALUATION_VALUE_LIST,
+            Queries.GET_EVALUATION_VALUE_LIST["variables"](
+                dataset_id=dataset_id,
+                prediction_set_id=prediction_set_id,
+                filter=filter_dict,
+                length=length,
+                cursor=cursor
+            )
+        )
+
+    def test_get_evaluation_value_list_without_filter(self):
+        """Test evaluation value list retrieval without filter."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        
+        mock_response = {
+            "evaluationValueList": {
+                "totalCount": 50,
+                "next": None,
+                "data": [{"dataId": "data-1"}]
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_evaluation_value_list(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id
+        )
+
+        # Assert
+        assert result["totalCount"] == 50
+        assert result["next"] is None
+        assert len(result["data"]) == 1
+
+    def test_get_evaluation_value_list_missing_dataset_id(self):
+        """Test evaluation value list with missing dataset_id."""
+        # Arrange
+        prediction_set_id = "pred-set-456"
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="dataset_id is required"):
+            self.data_service.get_evaluation_value_list(
+                dataset_id=None,
+                prediction_set_id=prediction_set_id
+            )
+
+    def test_get_evaluation_value_list_missing_prediction_set_id(self):
+        """Test evaluation value list with missing prediction_set_id."""
+        # Arrange
+        dataset_id = "dataset-123"
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="prediction_set_id is required"):
+            self.data_service.get_evaluation_value_list(
+                dataset_id=dataset_id,
+                prediction_set_id=None
+            )
+
+    def test_get_evaluation_value_list_empty_response(self):
+        """Test evaluation value list with empty response."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        
+        mock_response = {}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_evaluation_value_list(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id
+        )
+
+        # Assert
+        assert result == {}
+
+    def test_get_total_data_id_count_in_evaluation_value_success(self):
+        """Test successful total count retrieval."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        filter_dict = {"score": {"min": 0.8}}
+        
+        mock_response = {
+            "evaluationValueList": {
+                "totalCount": 150,
+                "next": "cursor-abc",
+                "data": [{"dataId": "data-1"}]
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_total_data_id_count_in_evaluation_value(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id,
+            filter=filter_dict
+        )
+
+        # Assert
+        assert result == 150
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.GET_EVALUATION_VALUE_LIST,
+            Queries.GET_EVALUATION_VALUE_LIST["variables"](
+                dataset_id=dataset_id,
+                prediction_set_id=prediction_set_id,
+                filter=filter_dict,
+                length=1
+            )
+        )
+
+    def test_get_total_data_id_count_in_evaluation_value_zero_count(self):
+        """Test total count retrieval with zero results."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        
+        mock_response = {
+            "evaluationValueList": {
+                "totalCount": 0,
+                "next": None,
+                "data": []
+            }
+        }
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_total_data_id_count_in_evaluation_value(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id
+        )
+
+        # Assert
+        assert result == 0
+
+    def test_get_total_data_id_count_in_evaluation_value_missing_response(self):
+        """Test total count retrieval with missing response fields."""
+        # Arrange
+        dataset_id = "dataset-123"
+        prediction_set_id = "pred-set-456"
+        
+        mock_response = {}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.get_total_data_id_count_in_evaluation_value(
+            dataset_id=dataset_id,
+            prediction_set_id=prediction_set_id
+        )
+
+        # Assert
+        assert result == 0

tests/datasets/__init__.py

@@ -0,0 +1 @@
+# Datasets test package

tests/datasets/test_dataset_service.py

@@ -0,0 +1,135 @@
+import pytest
+from unittest.mock import Mock
+
+from spb_onprem.datasets.service import DatasetService
+from spb_onprem.datasets.queries import Queries
+from spb_onprem.exceptions import BadParameterError
+
+
+class TestDatasetService:
+    """Test cases for DatasetService delete_dataset method."""
+
+    def setup_method(self):
+        """Set up test fixtures before each test method."""
+        self.dataset_service = DatasetService()
+        self.dataset_service.request_gql = Mock()
+
+    def test_delete_dataset_success(self):
+        """Test successful dataset deletion."""
+        # Arrange
+        dataset_id = "dataset-123"
+        mock_response = {"deleteDataset": True}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.dataset_service.delete_dataset(dataset_id)
+
+        # Assert
+        assert result is True
+        self.dataset_service.request_gql.assert_called_once_with(
+            Queries.DELETE_DATASET,
+            Queries.DELETE_DATASET["variables"](dataset_id=dataset_id)
+        )
+
+    def test_delete_dataset_failure(self):
+        """Test dataset deletion failure."""
+        # Arrange
+        dataset_id = "nonexistent-dataset"
+        mock_response = {"deleteDataset": False}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.dataset_service.delete_dataset(dataset_id)
+
+        # Assert
+        assert result is False
+        self.dataset_service.request_gql.assert_called_once_with(
+            Queries.DELETE_DATASET,
+            Queries.DELETE_DATASET["variables"](dataset_id=dataset_id)
+        )
+
+    def test_delete_dataset_missing_response(self):
+        """Test dataset deletion with missing response field."""
+        # Arrange
+        dataset_id = "dataset-123"
+        mock_response = {}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.dataset_service.delete_dataset(dataset_id)
+
+        # Assert
+        assert result is False
+
+    def test_delete_dataset_missing_dataset_id(self):
+        """Test delete dataset with missing dataset_id."""
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="dataset_id is required"):
+            self.dataset_service.delete_dataset(dataset_id=None)
+
+    def test_delete_dataset_empty_string_id(self):
+        """Test delete dataset with empty string ID."""
+        # Arrange
+        dataset_id = ""
+
+        # Act & Assert
+        # Empty string should be allowed by the service, but would fail at GraphQL level
+        mock_response = {"deleteDataset": False}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        result = self.dataset_service.delete_dataset(dataset_id)
+        assert result is False
+
+    def test_delete_dataset_query_structure(self):
+        """Test that delete dataset uses correct query structure."""
+        # Arrange
+        dataset_id = "test-dataset"
+        mock_response = {"deleteDataset": True}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        # Act
+        self.dataset_service.delete_dataset(dataset_id)
+
+        # Assert - Verify the query structure
+        call_args = self.dataset_service.request_gql.call_args
+        query = call_args[0][0]  # First positional argument
+        variables = call_args[0][1]  # Second positional argument
+
+        # Check query contains expected mutation
+        assert query["name"] == "deleteDataset"
+        assert "mutation DeleteDataset" in query["query"]
+        assert "deleteDataset(datasetId: $dataset_id)" in query["query"]
+
+        # Check variables function is called correctly
+        expected_variables = {"dataset_id": dataset_id}
+        assert variables == expected_variables
+
+    def test_delete_dataset_variables_function(self):
+        """Test that variables function works correctly."""
+        # Arrange
+        dataset_id = "test-variables-dataset"
+
+        # Act - Call the variables function directly
+        variables = Queries.DELETE_DATASET["variables"](dataset_id)
+
+        # Assert
+        expected_variables = {"dataset_id": dataset_id}
+        assert variables == expected_variables
+
+    def test_delete_dataset_with_special_characters(self):
+        """Test dataset deletion with special characters in ID."""
+        # Arrange
+        dataset_id = "dataset-with-special-chars_123@test"
+        mock_response = {"deleteDataset": True}
+        self.dataset_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.dataset_service.delete_dataset(dataset_id)
+
+        # Assert
+        assert result is True
+        
+        # Verify the special characters are passed through correctly
+        call_args = self.dataset_service.request_gql.call_args
+        variables = call_args[0][1]
+        assert variables["dataset_id"] == dataset_id

tests/exports/test_service.py

@@ -378,6 +378,7 @@ class TestExportService:
             dataset_id="test_dataset_id",
             name=Undefined,
             data_filter=Undefined,
+            location=Undefined,
             data_count=Undefined,
             frame_count=Undefined,
             annotation_count=Undefined,

tests/models/__init__.py

@@ -0,0 +1 @@
+# Models test package