From 70f4b1cdf7895de6a119dbe0b2379f4ed382ebb4 Mon Sep 17 00:00:00 2001 From: Benjamin Smith Date: Thu, 26 Dec 2024 09:19:47 -0500 Subject: [PATCH] Fix DB test --- tests/unit/destinations_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/destinations_test.py b/tests/unit/destinations_test.py index 5d2a90d..f74c179 100644 --- 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"},