Implemented test_to_config

Signed-off-by: Elena Khaustova <[email protected]>
kedro-org · Nov 14, 2024 · 0d9d241 · 0d9d241
1 parent f183e60
commit 0d9d241
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/tests/io/conftest.py b/tests/io/conftest.py
@@ -51,6 +51,27 @@ def correct_config(filepath):
     }
 
 
+@pytest.fixture
+def correct_config_versioned(filepath):
+    return {
+        "catalog": {
+            "boats": {
+                "type": "pandas.CSVDataset",
+                "filepath": filepath,
+                "versioned": True,
+            },
+            "cars": {
+                "type": "pandas.CSVDataset",
+                "filepath": "s3://test_bucket/test_file.csv",
+                "credentials": "cars_credentials",
+            },
+        },
+        "credentials": {
+            "cars_credentials": {"key": "FAKE_ACCESS_KEY", "secret": "FAKE_SECRET_KEY"}
+        },
+    }
+
+
 @pytest.fixture
 def correct_config_with_nested_creds(correct_config):
     correct_config["catalog"]["cars"]["credentials"] = {

diff --git a/tests/io/test_kedro_data_catalog.py b/tests/io/test_kedro_data_catalog.py
@@ -293,6 +293,32 @@ def test_release(self, data_catalog):
         """Test release is called without errors"""
         data_catalog.release("test")
 
+    class TestKedroDataCatalogToConfig:
+        def test_to_config(self, correct_config_versioned, dataset, filepath):
+            config = correct_config_versioned["catalog"]
+            credentials = correct_config_versioned["credentials"]
+            catalog = KedroDataCatalog.from_config(config, credentials)
+            catalog["resolved_ds"] = dataset
+
+            catalog_config, catalog_credentials, load_version, save_version = (
+                catalog.to_config()
+            )
+
+            expected_config = {
+                "resolved_ds": {
+                    "type": "kedro_datasets.pandas.csv_dataset.CSVDataset",
+                    "filepath": filepath,
+                    "save_args": {"index": False},
+                    "load_args": None,
+                    "credentials": None,
+                    "fs_args": None,
+                }
+            }
+            expected_config.update(config)
+
+            assert catalog_config == expected_config
+            assert catalog_credentials == credentials
+
     class TestKedroDataCatalogFromConfig:
         def test_from_correct_config(self, data_catalog_from_config, dummy_dataframe):
             """Test populating the data catalog from config"""