def check_load_cached_dataset(name):
# Test the caching using a temporary file.
with tempfile.TemporaryDirectory() as tmpdir:
# download and cache
ds = load_dataset(name, cache=True, data_home=tmpdir)
# use cached version
ds2 = load_dataset(name, cache=True, data_home=tmpdir)
assert_frame_equal(ds, ds2)
def test_load_dataset_passed_data_error():
df = pd.DataFrame()
err = "This function accepts only strings"
with pytest.raises(TypeError, match=err):
load_dataset(df)
def test_load_dataset_string_error():
name = "bad_name"
err = f"'{name}' is not one of the example datasets."
with pytest.raises(ValueError, match=err):
load_dataset(name)
def check_load_dataset(name):
ds = load_dataset(name, cache=False)
assert (isinstance(ds, pd.DataFrame))