Skip to content

Commit

Permalink
make more complete
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcoGorelli committed Aug 1, 2023
1 parent 7eae3e5 commit c0cdcca
Show file tree
Hide file tree
Showing 5 changed files with 36 additions and 11 deletions.
Binary file modified .coverage
Binary file not shown.
3 changes: 0 additions & 3 deletions check_completeness.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,9 +47,6 @@
# namespace
exclude = {
"Mapping",
"Column",
"DataFrame",
"GroupBy",
"column_object",
"dataframe_object",
"groupby_object",
Expand Down
5 changes: 5 additions & 0 deletions dataframe_api_compat/pandas_standard/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from dataframe_api_compat.pandas_standard.pandas_standard import (
PandasDataFrame,
PandasColumn,
PandasGroupBy,
)
import pandas as pd

Expand All @@ -10,6 +11,10 @@
Sequence,
)

Column = PandasColumn
DataFrame = PandasDataFrame
GroupBy = PandasGroupBy


class Int64:
...
Expand Down
5 changes: 5 additions & 0 deletions dataframe_api_compat/polars_standard/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,16 @@
from dataframe_api_compat.polars_standard.polars_standard import (
PolarsDataFrame,
PolarsColumn,
PolarsGroupBy,
)

import polars as pl
from typing import Any, Sequence

Column = PolarsColumn
DataFrame = PolarsDataFrame
GroupBy = PolarsGroupBy


class Int64:
...
Expand Down
34 changes: 26 additions & 8 deletions t.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,29 @@
import pandas as pd
import polars as pl
from datetime import date
from typing import Any
import dataframe_api_compat

df = pl.DataFrame(
{
"date": pl.date_range(date(2020, 10, 30), date(2021, 6, 1), eager=True),
}
)
df = df.with_columns(values=pl.arange(0, len(df)))

print(df.groupby_dynamic("date", every="3mo").agg(pl.col("values").sum()))
def convert_to_standard_compliant_dataframe(df: pd.DataFrame | pl.DataFrame) -> Any:
# todo: type return
if isinstance(df, pd.DataFrame):
return (
dataframe_api_compat.pandas_standard.convert_to_standard_compliant_dataframe(
df
)
)
elif isinstance(df, pl.DataFrame):
return (
dataframe_api_compat.polars_standard.convert_to_standard_compliant_dataframe(
df
)
)
else:
raise AssertionError(f"Got unexpected type: {type(df)}")


dfpd = pd.DataFrame({"a": [1, 2, 3]})
dfpl = pl.DataFrame({"a": [1, 2, 3]})

dfpd = convert_to_standard_compliant_dataframe(dfpd)
dfpl = convert_to_standard_compliant_dataframe(dfpl)

0 comments on commit c0cdcca

Please sign in to comment.