Skip to content

Commit 367d11a

Browse files
committed
update pandas tests to work with v3
1 parent f9a5344 commit 367d11a

6 files changed

Lines changed: 12 additions & 10 deletions

File tree

tests/tests_unit/test_api/test_raw.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -652,7 +652,8 @@ def test_dbs_to_pandas(self) -> None:
652652
pd.DataFrame({"name": ["kar", "car", "dar"]}), db_list.to_pandas().drop("created_time", axis=1)
653653
)
654654
pd.testing.assert_frame_equal(
655-
pd.DataFrame({"value": ["kar"]}, index=["name"]), db_list[0].to_pandas().drop("created_time")
655+
pd.DataFrame({"value": ["kar"]}, index=["name"], dtype=object),
656+
db_list[0].to_pandas().drop("created_time"),
656657
)
657658

658659
def test_tables_to_pandas(self) -> None:
@@ -666,7 +667,8 @@ def test_tables_to_pandas(self) -> None:
666667
pd.DataFrame({"name": ["kar", "car", "dar"]}), table_list.to_pandas().drop("created_time", axis=1)
667668
)
668669
pd.testing.assert_frame_equal(
669-
pd.DataFrame({"value": ["kar"]}, index=["name"]), table_list[0].to_pandas().drop("created_time")
670+
pd.DataFrame({"value": ["kar"]}, index=["name"], dtype=object),
671+
table_list[0].to_pandas().drop("created_time"),
670672
)
671673

672674
def test_rows_to_pandas(self) -> None:

tests/tests_unit/test_api/test_sequences.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -740,8 +740,8 @@ def test_sequences_to_pandas(self, cognite_client: CogniteClient, mock_seq_respo
740740
df = seq.to_pandas(expand_metadata=True, metadata_prefix="")
741741
assert isinstance(df, pd.DataFrame)
742742
assert "metadata" not in df.columns
743-
assert "string" == df.loc["description"][0]
744-
assert "metadata-value" == df.loc["metadata-key"][0]
743+
assert "string" == df.loc["description"].iloc[0]
744+
assert "metadata-value" == df.loc["metadata-key"].iloc[0]
745745

746746
def test_insert_dataframe_extids(self, cognite_client: CogniteClient, mock_post_sequence_data: HTTPXMock) -> None:
747747
import pandas as pd

tests/tests_unit/test_api/test_time_series.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,5 +259,5 @@ def test_time_series_to_pandas(self, cognite_client: CogniteClient, mock_ts_resp
259259
df = ts.to_pandas(expand_metadata=True, metadata_prefix="", camel_case=True)
260260
assert isinstance(df, pd.DataFrame)
261261
assert "metadata" not in df.columns
262-
assert [0] == df.loc["securityCategories"][0]
263-
assert "metadata-value" == df.loc["metadata-key"][0]
262+
assert [0] == df.loc["securityCategories"].iloc[0]
263+
assert "metadata-value" == df.loc["metadata-key"].iloc[0]

tests/tests_unit/test_data_classes/test_data_models/test_typed_instances.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -305,8 +305,8 @@ def test_to_pandas_list(self, person_read: PersonRead) -> None:
305305
"space": ["sp_my_fixed_space"],
306306
"external_id": ["my_external_id"],
307307
"version": [1],
308-
"last_updated_time": [pd.Timestamp("1970-01-01 00:00:00")],
309-
"created_time": [pd.Timestamp("1970-01-01 00:00:00")],
308+
"last_updated_time": [pd.Timestamp(0, unit="ms")],
309+
"created_time": [pd.Timestamp(0, unit="ms")],
310310
"instance_type": ["node"],
311311
"type": [{"space": "sp_model_space", "external_id": "person"}],
312312
"name": ["John Doe"],

tests/tests_unit/test_data_classes/test_datapoints.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def test_identifier_priority(self, dps_lst_cls: type[CogniteResourceList]) -> No
111111

112112
exp_df = pd.DataFrame(
113113
{1: 2.0, 2: 4.0, 3: 6.0},
114-
index=np.array([1234 * 1_000_000], dtype="datetime64[ns]"),
114+
index=np.array([1234], dtype="datetime64[ms]"),
115115
)
116116
exp_df.columns = pd.MultiIndex.from_tuples(
117117
[(123,), ("foo",), (NodeId(space="s", external_id="x"),)], names=["identifier"]

tests/tests_unit/test_data_classes/test_groups.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ class TestGroupsList:
104104
@pytest.mark.parametrize(
105105
"convert_timestamps, expected",
106106
(
107-
(True, dict(data=[None, "1970-01-02 10:17:36.789", None], dtype="datetime64[ns]", name="deleted_time")),
107+
(True, dict(data=[None, "1970-01-02 10:17:36.789", None], dtype="datetime64[ms]", name="deleted_time")),
108108
(False, dict(data=[-1, 123456789, None], dtype="Int64", name="deleted_time")),
109109
),
110110
)

0 commit comments

Comments
 (0)