Skip to content

Commit b0ccd3a

Browse files
authored
fix unit test (#3127)
1 parent a26bf1e commit b0ccd3a

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

darts/tests/test_timeseries.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,11 @@ def test_creation_from_dataframe_with_dates(self, backend, date_type):
203203

204204
# internally, Darts converts any temporal dtype into Datetime
205205
ts = TimeSeries.from_dataframe(df, time_col="time")
206-
assert ts.time_index.equals(pd.DatetimeIndex(data["time"], name="time"))
206+
assert ts.time_index.equals(
207+
pd.DatetimeIndex(
208+
data["time"], name="time", freq="MS", dtype=ts.time_index.dtype
209+
)
210+
)
207211
np.testing.assert_array_equal(
208212
ts.values(), np.array(data["values"])[:, np.newaxis]
209213
)

0 commit comments

Comments
 (0)