def test_tables_from_stack(): c = Context() assert not c._get_tables_from_stack() df = pd.DataFrame() assert "df" in c._get_tables_from_stack() def f(): df2 = pd.DataFrame() assert "df" in c._get_tables_from_stack() assert "df2" in c._get_tables_from_stack() f() def g(): df = pd.DataFrame({"a": [1]}) assert "df" in c._get_tables_from_stack() assert c._get_tables_from_stack()["df"].columns == ["a"]
def test_tables_from_stack(gpu): c = Context() assert not c._get_tables_from_stack() df = pd.DataFrame() if not gpu else cudf.DataFrame() assert "df" in c._get_tables_from_stack() def f(gpu): df2 = pd.DataFrame() if not gpu else cudf.DataFrame() assert "df" in c._get_tables_from_stack() assert "df2" in c._get_tables_from_stack() f(gpu=gpu) def g(gpu=gpu): df = pd.DataFrame({"a": [1]}) if not gpu else cudf.DataFrame({"a": [1]}) assert "df" in c._get_tables_from_stack() assert c._get_tables_from_stack()["df"].columns == ["a"] g(gpu=gpu)