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