def test_get_first_datapoint(self, mock_get_first_dp_in_ts):
now = utils.timestamp_to_ms("now")
res = TS_CLIENT.retrieve(id=1).first()
assert isinstance(res, Datapoint)
assert Datapoint(timestamp=1, value=10) == res
assert 0 == jsgz_load(mock_get_first_dp_in_ts.calls[1].request.body)["start"]
assert now <= jsgz_load(mock_get_first_dp_in_ts.calls[1].request.body)["end"]
assert 1 == jsgz_load(mock_get_first_dp_in_ts.calls[1].request.body)["limit"]
def test_get_count(self, mock_count_dps_in_ts):
now = utils.timestamp_to_ms("now")
assert 15 == TS_CLIENT.retrieve(id=1).count()
assert "count" == jsgz_load(
mock_count_dps_in_ts.calls[1].request.body)["aggregates"][0]
assert 0 == jsgz_load(
mock_count_dps_in_ts.calls[1].request.body)["start"]
assert now <= jsgz_load(
mock_count_dps_in_ts.calls[1].request.body)["end"]
def test_partitioned_list(self):
# stop race conditions by cutting off max created time
maxtime = utils.timestamp_to_ms(datetime(2019, 5, 25, 17, 30))
res_flat = COGNITE_CLIENT.events.list(limit=None, type="test-data-populator", start_time={"max": maxtime})
res_part = COGNITE_CLIENT.events.list(
partitions=8, type="test-data-populator", start_time={"max": maxtime}, limit=None
)
assert len(res_flat) > 0
assert len(res_flat) == len(res_part)
assert {a.id for a in res_flat} == {a.id for a in res_part}
def test_asset(self, mock_ts_by_ids_response, mock_asset_by_ids_response):
now = utils.timestamp_to_ms("now")
asset = TS_CLIENT.retrieve(id=1).asset()
assert isinstance(asset, Asset)
assert "assetname" == asset.name
