def test_undefined_dataset():
with pytest.raises(AttributeError) as err:
data("blahblahblah")
assert str(err.value) == "No dataset named 'blahblahblah'"
with pytest.raises(AttributeError) as err:
local_data("blahblahblah")
assert str(err.value) == "No dataset named 'blahblahblah'"
def test_load_local_dataset(name):
loader = getattr(data, name.replace("-", "_"))
local_loader = getattr(local_data, name.replace("-", "_"))
df1 = data(name)
df2 = loader() # equivalent to data.dataset_name()
df3 = local_data(name)
df4 = local_loader() # equivalent to local_data.dataset_name()
assert_frame_equal(df1, df2)
assert_frame_equal(df1, df3)
assert_frame_equal(df1, df4)
raw1 = loader.raw()
raw2 = local_loader.raw()
raw3 = data(name, return_raw=True)
raw4 = local_data(name, return_raw=True)
assert raw1 == raw2 == raw3 == raw4
assert type(raw1) is type(raw2) is type(raw3) is type(raw4) is bytes
def test_local_dataset_error(name):
with pytest.raises(ValueError) as err:
local_data(name)
assert str(err.value).startswith("'{0}' dataset is not available locally"
"".format(name.replace("-", "_")))