Add folder level

sdv-dev · Apr 25, 2024 · 134b160 · 134b160
1 parent 22414b5
commit 134b160
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 2 deletions.
diff --git a/sdv/io/local/__init__.py b/sdv/io/local/__init__.py
@@ -0,0 +1,8 @@
+"""Local I/O module."""
+
+from sdv.io.local.local import BaseLocalHandler, CSVHandler
+
+__all__ = (
+    'BaseLocalHandler',
+    'CSVHandler'
+)
diff --git a/sdv/io/local.py → sdv/io/local/local.py b/sdv/io/local.py → sdv/io/local/local.py
diff --git a/tests/integration/io/local/__init__.py b/tests/integration/io/local/__init__.py
diff --git a/tests/integration/io/test_local.py → tests/integration/io/local/test_local.py b/tests/integration/io/test_local.py → tests/integration/io/local/test_local.py
diff --git a/tests/unit/io/local/__init__.py b/tests/unit/io/local/__init__.py
diff --git a/tests/unit/io/test_local.py → tests/unit/io/local/test_local.py b/tests/unit/io/test_local.py → tests/unit/io/local/test_local.py
@@ -6,7 +6,7 @@
 import pandas as pd
 import pytest
 
-from sdv.io.local import CSVHandler
+from sdv.io.local.local import CSVHandler
 from sdv.metadata.multi_table import MultiTableMetadata
 
 
@@ -41,7 +41,7 @@ def test___init___error_encoding(self):
         with pytest.raises(ValueError, match=error_msg):
             CSVHandler(sep=';', encoding='sdvutf-8', decimal=',', float_format='%.2f')
 
-    @patch('sdv.io.local.Path.glob')
+    @patch('sdv.io.local.local.Path.glob')
     @patch('pandas.read_csv')
     def test_read(self, mock_read_csv, mock_glob):
         """Test the read method of CSVHandler class with a folder."""