update test

cognitedata · May 30, 2024 · 4db0b81 · 4db0b81
1 parent e7b43f7
commit 4db0b81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tests_integration/test_api/test_raw.py b/tests/tests_integration/test_api/test_raw.py
@@ -126,7 +126,7 @@ def test_insert_and_retrieve_dataframe(self, cognite_client, new_database_with_t
         data = {"a": {"r1": 1, "r2": 1, "r3": 1}, "b": {"r1": None, "r2": None, "r3": None}}
 
         df = pd.DataFrame.from_dict(data)
-        cognite_client.raw.rows.insert_dataframe(db.name, table.name, df)
+        cognite_client.raw.rows.insert_dataframe(db.name, table.name, df, dropna=False)
         retrieved_df = cognite_client.raw.rows.retrieve_dataframe(db.name, table.name)
 
         pd.testing.assert_frame_equal(df.sort_index(), retrieved_df.sort_index())