add load_args check

kedro-org · Dec 16, 2024 · 4ca50fc · 4ca50fc
1 parent db8971b
commit 4ca50fc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/kedro-datasets/tests/ibis/test_table_dataset.py b/kedro-datasets/tests/ibis/test_table_dataset.py
@@ -100,6 +100,11 @@ def test_exists(self, table_dataset, dummy_table):
         table_dataset.save(dummy_table)
         assert table_dataset.exists()
 
+    def test_load_extra_params(self):
+        """Test overriding the load arguments."""
+        table_dataset = TableDataset(table_name="test", load_args={"database": "test"})
+        assert "database" in table_dataset._load_args
+
     @pytest.mark.parametrize("save_args", [{"materialized": "table"}], indirect=True)
     def test_save_extra_params(self, table_dataset, save_args, dummy_table, database):
         """Test overriding the default save arguments."""