Python CogniteClient Examples

Example #1

File: cdf.py Project: cognitedata/powerdummy-populator

def connect():
    return CogniteClient(
        project="powerdummy",
        api_key=os.environ["POWERDUMMY_API_KEY"],
        base_url="https://greenfield.cognitedata.com",
        client_name="powerdummy-populator",
    )

Example #2

File: test_data_sets.py Project: pubuduSiti/cognite-sdk-python

import re

import pytest

from cognite.client.data_classes import DataSet, DataSetList, DataSetUpdate, TimestampRange
from cognite.client.experimental import CogniteClient
from tests.utils import jsgz_load

# from unittest import mock


COGNITE_CLIENT = CogniteClient()
DS_API = COGNITE_CLIENT.data_sets


@pytest.fixture
def dataset_update():
    yield DataSetUpdate(id=1).description.set("_")


@pytest.fixture
def mock_ds_response(rsps):
    response_body = {
        "items": [
            {
                "id": 0,
                "externalId": "string",
                "name": "string",
                "description": "string",
                "metadata": {"metadata-key": "metadata-value"},
                "writeProtected": False,

Example #3

from unittest import mock
from unittest.mock import call

import pytest

from cognite.client.data_classes import Asset, AssetList, Event, EventList, FileMetadata, FileMetadataList
from cognite.client.experimental import CogniteClient

c = CogniteClient()


class TestAsset:
    def test_get_events(self):
        c.events.list = mock.MagicMock()
        a = Asset(id=1, cognite_client=c)
        a.events()
        assert c.events.list.call_args == call(asset_ids=[1])
        assert c.events.list.call_count == 1

    def test_get_time_series(self):
        c.time_series.list = mock.MagicMock()
        a = Asset(id=1, cognite_client=c)
        a.time_series()
        assert c.time_series.list.call_args == call(asset_ids=[1])
        assert c.time_series.list.call_count == 1

    def test_get_sequences(self):
        c.sequences.list = mock.MagicMock()
        a = Asset(id=1, cognite_client=c)
        a.sequences()
        assert c.sequences.list.call_args == call(asset_ids=[1])

Example #4

File: generate_code_snippets.py Project: pubuduSiti/cognite-sdk-python

from cognite.client._api_client import APIClient
from cognite.client.experimental import CogniteClient


def collect_apis(obj, done):
    if done.get(obj.__class__):
        return []
    done[obj.__class__] = True
    apis = inspect.getmembers(
        obj, lambda m: isinstance(m, APIClient) and not done.get(m.__class__))
    sub = [(n + "." + sn, sa) for n, c in apis
           for sn, sa in collect_apis(c, done)]
    return apis + sub


client = CogniteClient(project="_", api_key="_", client_name="_")
parser = DocTestParser()

apis = collect_apis(client, {})

snippets = {
    "language": "Python",
    "label": "Python SDK",
    "operations": defaultdict(str)
}
filter_out = [
    "from cognite.client import CogniteClient", "c = CogniteClient()", ""
]

duplicate_operations = {
    "listAssets": "getAssets",

Example #5

File: test_models.py Project: ik-learning/cognite-sdk-python

import pytest

from cognite.client._api.model_hosting.models import EmptyArtifactsDirectory, PredictionError
from cognite.client.data_classes.model_hosting.models import (
    Model,
    ModelArtifact,
    ModelArtifactList,
    ModelList,
    ModelVersion,
    ModelVersionList,
    ModelVersionLog,
)
from cognite.client.experimental import CogniteClient
from tests.utils import jsgz_load

MODELS_API = CogniteClient().model_hosting.models


class TestModels:
    @pytest.fixture
    def created_model(self):
        model_name = "test-model-{}".format(randint(0, 1e5))
        model = MODELS_API.create_model(name=model_name, webhook_url="https://bla.bla")
        yield model
        MODELS_API.delete_model(model.id)

    @pytest.fixture
    def mock_online_predict_ok(self, rsps):
        rsps.add(
            rsps.PUT,
            MODELS_API._get_base_url_with_base_path() + "/analytics/models/1/predict",

Example #6

File: test_schedules.py Project: ik-learning/cognite-sdk-python

from unittest import mock

import pytest

from cognite.client.data_classes.model_hosting.schedules import Schedule, ScheduleList, ScheduleLog
from cognite.client.experimental import CogniteClient
from tests.utils import jsgz_load

SCHEDULES_API = CogniteClient().model_hosting.schedules


class TestSchedules:
    schedule_response = {
        "data": {
            "items": [{
                "isDeprecated": False,
                "name": "test-schedule",
                "dataSpec": {
                    "spec": "spec"
                },
                "modelId": 123,
                "createdTime": 0,
                "metadata": {
                    "k": "v"
                },
                "id": 123,
                "args": {
                    "k": "v"
                },
                "description": "string",
            }]

Example #7

File: test_source_packages.py Project: lightspin/cognite-sdk-python

import os
import time
from random import randint
from tempfile import TemporaryDirectory

import pytest

from cognite.client.data_classes.model_hosting.source_packages import SourcePackage, SourcePackageList
from cognite.client.exceptions import CogniteAPIError
from cognite.client.experimental import CogniteClient

SOURCE_PACKAGES_API = CogniteClient().model_hosting.source_packages


class TestSourcePackages:
    @pytest.fixture(scope="class")
    def source_package_file_path(self):
        with TemporaryDirectory() as tmp_dir:
            file_path = os.path.join(tmp_dir, "sp.tar.gz")
            with open(file_path, "w") as f:
                f.write("content")
            yield file_path

    @pytest.fixture(scope="class")
    def created_source_package(self, source_package_file_path):
        sp_name = "test-sp-{}".format(randint(0, 1e5))
        sp = SOURCE_PACKAGES_API.upload_source_package(
            name=sp_name,
            package_name="whatever",
            available_operations=["TRAIN", "PREDICT"],
            runtime_version="0.1",

Example #8

File: test_versions.py Project: lightspin/cognite-sdk-python

import pytest

from cognite.client._api.model_hosting.models import PredictionError
from cognite.client._api.model_hosting.versions import EmptyArtifactsDirectory
from cognite.client.data_classes.model_hosting.versions import (
    ModelArtifact,
    ModelArtifactList,
    ModelVersion,
    ModelVersionList,
    ModelVersionLog,
)
from cognite.client.experimental import CogniteClient
from tests.utils import jsgz_load

MODELS_API = CogniteClient().model_hosting.models
VERSIONS_API = CogniteClient().model_hosting.versions


class TestVersions:
    model_version_response = {
        "isDeprecated": True,
        "trainingDetails": {
            "machineType": "string",
            "scaleTier": "string",
            "completedTime": 0,
            "sourcePackageId": "string",
            "args": "string",
        },
        "name": "version1",
        "errorMsg": "string",