Updated tests for removed datasets

Author: fealho

tests/integration/datasets/test_demo.py

@@ -1,6 +1,6 @@
 import pandas as pd
 
-from sdv.datasets.demo import get_available_demos, get_readme, get_source
+from sdv.datasets.demo import get_available_demos
 
 
 def test_get_available_demos_single_table():
@@ -85,30 +85,3 @@ def test_get_available_demos_multi_table():
         'num_tables': [2, 2],
     })
     pd.testing.assert_frame_equal(tables_info[['dataset_name', 'size_MB', 'num_tables']], expected)
-
-
-def test_get_readme_and_source_single_table_dataset1(tmp_path):
-    """Test it returns the README and SOURCE for a single table dataset."""
-    # Run
-    readme = get_readme('single_table', 'dataset1')
-    source = get_source('single_table', 'dataset1')
-
-    # Assert
-    assert isinstance(readme, str) and 'sample dataset' in readme.lower()
-    assert isinstance(source, str) and source.strip() == 'unknown'
-
-    readme_out = tmp_path / 'r.txt'
-    source_out = tmp_path / 's.txt'
-    readme2 = get_readme('single_table', 'dataset1', str(readme_out))
-    source2 = get_source('single_table', 'dataset1', str(source_out))
-    assert readme2 == readme
-    assert source2 == source
-    assert readme_out.read_text(encoding='utf-8').strip() == readme.strip()
-    assert source_out.read_text(encoding='utf-8').strip() == source.strip()
-
-
-def test_get_readme_missing_returns_none():
-    """Test it returns None when the README/SOURCE is missing."""
-    # Run and Assert
-    assert get_readme('single_table', 'dataset2') is None
-    assert get_source('single_table', 'dataset2') is None

tests/unit/datasets/test_demo.py

@@ -342,18 +342,16 @@ def test_get_available_demos_invalid_modality():
 def test__find_data_zip_key():
     # Setup
     contents = [
-        {'Key': 'single_table/dataset/data.ZIP'},
-        {'Key': 'single_table/dataset/metadata.json'},
-        {'Key': 'single_table/dataset/aaa_wrong.json'},
-        {'Key': 'single_table/dataset/README.txt'},
+        {'Key': 'single_table/fake_hotel_guests/data.ZIP'},
+        {'Key': 'single_table/fake_hotel_guests/metadata.json'},
     ]
-    dataset_prefix = 'single_table/dataset/'
+    dataset_prefix = 'single_table/fake_hotel_guests/'
 
     # Run
     zip_key = _find_data_zip_key(contents, dataset_prefix)
 
     # Assert
-    assert zip_key == 'single_table/dataset/data.ZIP'
+    assert zip_key == 'single_table/fake_hotel_guests/data.ZIP'
 
 
 @patch('sdv.datasets.demo._get_data_from_bucket')