Add integration tests and dependencies to read and write xslx

Makefile

@@ -261,5 +261,5 @@ release-major: check-release bumpversion-major release
 
 .PHONY: check-deps
 check-deps:
-	$(eval allow_list='cloudpickle=|graphviz=|numpy=|pandas=|tqdm=|copulas=|ctgan=|deepecho=|rdt=|sdmetrics=|platformdirs=')
+	$(eval allow_list='cloudpickle=|graphviz=|numpy=|pandas=|tqdm=|copulas=|ctgan=|deepecho=|rdt=|sdmetrics=|platformdirs=|openpyxl=|xlsxwriter=')
 	pip freeze | grep -v "SDV.git" | grep -E $(allow_list) | sort > $(OUTPUT_FILEPATH)

pyproject.toml

@@ -38,7 +38,9 @@ dependencies = [
     'deepecho>=0.6.0',
     'rdt>=1.12.0',
     'sdmetrics>=0.14.0',
-    'platformdirs>=4.0'
+    'platformdirs>=4.0',
+    'xlsxwriter>=3.1.0',
+    'openpyxl>=3.1.0'
 ]
 
 [project.urls]

sdv/io/local/__init__.py

@@ -1,8 +1,9 @@
 """Local I/O module."""
 
-from sdv.io.local.local import BaseLocalHandler, CSVHandler
+from sdv.io.local.local import BaseLocalHandler, CSVHandler, ExcelHandler
 
 __all__ = (
     'BaseLocalHandler',
-    'CSVHandler'
+    'CSVHandler',
+    'ExcelHandler'
 )

tests/integration/io/local/test_local.py

@@ -1,13 +1,13 @@
 import pandas as pd
 
-from sdv.io.local import CSVHandler
+from sdv.io.local import CSVHandler, ExcelHandler
 from sdv.metadata import MultiTableMetadata
 
 
 class TestCSVHandler:
 
-    def test_integration_read_write(self, tmpdir):
-        """Test end to end the read and write methods of ``CSVHandler``."""
+    def test_integration_write_and_read(self, tmpdir):
+        """Test end to end the write and read methods of ``CSVHandler``."""
         # Prepare synthetic data
         synthetic_data = {
             'table1': pd.DataFrame({'col1': [1, 2, 3], 'col2': ['a', 'b', 'c']}),
@@ -30,3 +30,31 @@ def test_integration_read_write(self, tmpdir):
         # Check if the dataframes match the original synthetic data
         pd.testing.assert_frame_equal(data['table1'], synthetic_data['table1'])
         pd.testing.assert_frame_equal(data['table2'], synthetic_data['table2'])
+
+
+class TestExcelHandler:
+
+    def test_integration_write_and_read(self, tmpdir):
+        """Test end to end the write and read methods of ``ExcelHandler``."""
+        # Prepare synthetic data
+        synthetic_data = {
+            'table1': pd.DataFrame({'col1': [1, 2, 3], 'col2': ['a', 'b', 'c']}),
+            'table2': pd.DataFrame({'col3': [4, 5, 6], 'col4': ['d', 'e', 'f']})
+        }
+
+        # Write synthetic data to xslx files
+        handler = ExcelHandler()
+        handler.write(synthetic_data, tmpdir / 'excel.xslx')
+
+        # Read data from xslx file
+        data, metadata = handler.read(tmpdir / 'excel.xslx')
+
+        # Check if data was read correctly
+        assert len(data) == 2
+        assert 'table1' in data
+        assert 'table2' in data
+        assert isinstance(metadata, MultiTableMetadata) is True
+
+        # Check if the dataframes match the original synthetic data
+        pd.testing.assert_frame_equal(data['table1'], synthetic_data['table1'])
+        pd.testing.assert_frame_equal(data['table2'], synthetic_data['table2'])