Fix DB test

bh2smith · Dec 26, 2024 · 70f4b1c · 70f4b1c
1 parent 1aab8a6
commit 70f4b1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/destinations_test.py b/tests/unit/destinations_test.py
@@ -39,7 +39,7 @@ def test_ensure_index_disabled_when_uploading(self, mock_to_csv, *_):
             table_name="foo",
             request_timeout=10,
         )
-        destination.save(dummy_df)
+        destination.save(TypedDataFrame(dummy_df, {}))
         mock_to_csv.assert_called_once_with(index=False)
 
     @patch("pandas.core.generic.NDFrame.to_csv", name="Fake csv writer")
@@ -55,7 +55,7 @@ def test_duneclient_sets_timeout(self, mock_to_csv, *_):
     @patch("dune_client.api.table.TableAPI.upload_csv", name="Fake CSV uploader")
     def test_dune_error_handling(self, mock_dune_upload_csv):
         dest = DuneDestination(api_key="f00b4r", table_name="foo", request_timeout=10)
-        df = pd.DataFrame([{"foo": "bar"}])
+        df = TypedDataFrame(pd.DataFrame([{"foo": "bar"}]), {})
 
         dune_err = DuneError(
             data={"error": "bad stuff"},