Move test to local instead of s3

Author: pvk-developer

tests/integration/test_benchmark.py

@@ -202,8 +202,7 @@ def test_benchmark_single_table_duplicate_synthesizers():
 
     # Run and Assert
     error_msg = re.escape(
-        'Synthesizers must be unique. Please remove repeated values in the `synthesizers` '
-        'and `custom_synthesizers` parameters.'
+        'Synthesizers must be unique. Please remove repeated values in the provided synthesizers.'
     )
     with pytest.raises(ValueError, match=error_msg):
         sdgym.benchmark_single_table(

tests/unit/test_result_writer.py

@@ -101,6 +101,37 @@ def test_write_yaml_append(self, tmp_path):
         expected_data = {**data1, **data2}
         assert loaded_data == expected_data
 
+    def test_write_zipped_dataframes(self, tmp_path):
+        """Test the `write_zipped_dataframes` method."""
+        # Setup
+        base_path = tmp_path / 'sdgym_results'
+        base_path.mkdir(parents=True, exist_ok=True)
+        result_writer = LocalResultsWriter()
+        file_path = base_path / 'data.zip'
+
+        data = {
+            'table1': pd.DataFrame({'a': [1, 2], 'b': [3, 4]}),
+            'table2': pd.DataFrame({'x': [5, 6], 'y': [7, 8]}),
+        }
+
+        # Run
+        result_writer.write_zipped_dataframes(data, file_path)
+
+        # Assert
+        assert file_path.exists()
+
+        with zipfile.ZipFile(file_path, 'r') as zf:
+            # Check that all tables are present
+            names = zf.namelist()
+            assert 'table1.csv' in names
+            assert 'table2.csv' in names
+
+            # Check each table content matches the original
+            for table_name, df in data.items():
+                with zf.open(f'{table_name}.csv') as f:
+                    loaded_df = pd.read_csv(f)
+                    pd.testing.assert_frame_equal(df, loaded_df)
+
 
 class TestS3ResultsWriter:
     def test__init__(self):
@@ -225,34 +256,3 @@ def test_write_yaml_append(self, mockparse_s3_path):
             Bucket='bucket_name',
             Key='key_prefix/test_data.yaml',
         )
-
-    def test_write_zipped_dataframes(self, tmp_path):
-        """Test the `write_zipped_dataframes` method."""
-        # Setup
-        base_path = tmp_path / 'sdgym_results'
-        base_path.mkdir(parents=True, exist_ok=True)
-        result_writer = LocalResultsWriter()
-        file_path = base_path / 'data.zip'
-
-        data = {
-            'table1': pd.DataFrame({'a': [1, 2], 'b': [3, 4]}),
-            'table2': pd.DataFrame({'x': [5, 6], 'y': [7, 8]}),
-        }
-
-        # Run
-        result_writer.write_zipped_dataframes(data, file_path)
-
-        # Assert
-        assert file_path.exists()
-
-        with zipfile.ZipFile(file_path, 'r') as zf:
-            # Check that all tables are present
-            names = zf.namelist()
-            assert 'table1.csv' in names
-            assert 'table2.csv' in names
-
-            # Check each table content matches the original
-            for table_name, df in data.items():
-                with zf.open(f'{table_name}.csv') as f:
-                    loaded_df = pd.read_csv(f)
-                    pd.testing.assert_frame_equal(df, loaded_df)