From cac0bac4265e4fd2aa05406651114c217f1f9a43 Mon Sep 17 00:00:00 2001 From: Drake Eidukas <drake@nominal.io> Date: Tue, 29 Oct 2024 14:18:22 -0700 Subject: [PATCH] Fix dataset tests --- tests/core/test_dataset.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/core/test_dataset.py b/tests/core/test_dataset.py index 5036027b..782e8701 100644 --- a/tests/core/test_dataset.py +++ b/tests/core/test_dataset.py @@ -1,12 +1,12 @@ import functools -from datetime import timedelta +from datetime import datetime, timedelta, timezone from unittest.mock import MagicMock, patch import pytest from nominal._api.combined.timeseries_logicalseries_api import BatchUpdateLogicalSeriesRequest from nominal.core.channel import Channel -from nominal.core.dataset import Dataset +from nominal.core.dataset import Dataset, DatasetBounds from nominal.core.unit import Unit from nominal.exceptions import NominalIngestError, NominalIngestFailed @@ -30,6 +30,10 @@ def mock_dataset(mock_clients): rid="test-rid", name="Test Dataset", description="A dataset for testing", + bounds=DatasetBounds( + start=datetime(year=2021, month=10, day=9, hour=10, minute=9, second=10, tzinfo=timezone.utc), + end=datetime(year=2021, month=10, day=9, hour=11, minute=9, second=10, tzinfo=timezone.utc), + ), properties={}, labels=[], _clients=mock_clients,