add mock

cognitedata · Jan 17, 2025 · 2bed274 · 2bed274
1 parent 7a0e15f
commit 2bed274
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cognite/client/testing.py b/cognite/client/testing.py
@@ -17,6 +17,7 @@
 from cognite.client._api.data_modeling.graphql import DataModelingGraphQLAPI
 from cognite.client._api.data_modeling.instances import InstancesAPI
 from cognite.client._api.data_modeling.spaces import SpacesAPI
+from cognite.client._api.data_modeling.statistics import StatisticsAPI
 from cognite.client._api.data_modeling.views import ViewsAPI
 from cognite.client._api.data_sets import DataSetsAPI
 from cognite.client._api.datapoints import DatapointsAPI
@@ -114,6 +115,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.data_modeling.views = MagicMock(spec_set=ViewsAPI)
         self.data_modeling.instances = MagicMock(spec_set=InstancesAPI)
         self.data_modeling.graphql = MagicMock(spec_set=DataModelingGraphQLAPI)
+        self.data_modeling.statistics = MagicMock(spec_set=StatisticsAPI)
 
         self.data_sets = MagicMock(spec_set=DataSetsAPI)