def test_deprecation_warning():
c = Context()
data_frame = dd.from_pandas(pd.DataFrame(), npartitions=1)
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
c.register_dask_table(data_frame, "table")
assert len(w) == 1
assert issubclass(w[-1].category, DeprecationWarning)
assert "table" in c.tables
c.drop_table("table")
assert "table" not in c.tables
def test_add_remove_tables():
c = Context()
data_frame = dd.from_pandas(pd.DataFrame(), npartitions=1)
c.create_table("table", data_frame)
assert "table" in c.tables
c.drop_table("table")
assert "table" not in c.tables
with pytest.raises(KeyError):
c.drop_table("table")
c.create_table("table", [data_frame])
assert "table" in c.tables
