diff --git a/tests/snippets/frame/__init__.py b/tests/snippets/frame/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/test_frame.py b/tests/snippets/frame/test_typing.py similarity index 100% rename from tests/snippets/test_frame.py rename to tests/snippets/frame/test_typing.py diff --git a/tests/snippets/io/__init__.py b/tests/snippets/io/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/io/formats/__init__.py b/tests/snippets/io/formats/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/io/formats/test_to_csv_typing.py b/tests/snippets/io/formats/test_to_csv_typing.py new file mode 100644 index 0000000..acf634d --- /dev/null +++ b/tests/snippets/io/formats/test_to_csv_typing.py @@ -0,0 +1,23 @@ +# flake8: noqa: F841 +import tempfile + +import pandas as pd +from pandas.io.parsers import TextFileReader + + +def test_types_read_csv() -> None: + df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]}) + csv_df: str = df.to_csv() + + with tempfile.NamedTemporaryFile() as file: + df.to_csv(file.name) + df2: pd.DataFrame = pd.read_csv(file.name) + df3: pd.DataFrame = pd.read_csv(file.name, sep="a", squeeze=False) + df4: pd.DataFrame = pd.read_csv(file.name, header=None, prefix="b", mangle_dupe_cols=True, keep_default_na=False) + df5: pd.DataFrame = pd.read_csv(file.name, engine='python', true_values=[0, 1, 3], na_filter=False) + df6: pd.DataFrame = pd.read_csv(file.name, skiprows=lambda x: x in [0, 2], skip_blank_lines=True, dayfirst=False) + df7: pd.DataFrame = pd.read_csv(file.name, nrows=2) + tfr1: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=True, chunksize=3) + tfr2: TextFileReader = pd.read_csv(file.name, nrows=2, chunksize=1) + tfr3: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=False, chunksize=1) + tfr4: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=True) diff --git a/tests/snippets/io/json/__init__.py b/tests/snippets/io/json/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/io/json/test_normalize_typing.py b/tests/snippets/io/json/test_normalize_typing.py new file mode 100644 index 0000000..2cc65d8 --- /dev/null +++ b/tests/snippets/io/json/test_normalize_typing.py @@ -0,0 +1,16 @@ +# flake8: noqa: F841 +from typing import Any, Dict, List + +import pandas as pd + + +def test_types_json_normalize() -> None: + data1: List[Dict[str, Any]] = [{'id': 1, 'name': {'first': 'Coleen', 'last': 'Volk'}}, + {'name': {'given': 'Mose', 'family': 'Regner'}}, + {'id': 2, 'name': 'Faye Raker'}] + df1: pd.DataFrame = pd.json_normalize(data=data1) + df2: pd.DataFrame = pd.json_normalize(data=data1, max_level=0, sep=";") + df3: pd.DataFrame = pd.json_normalize(data=data1, meta_prefix="id", record_prefix="name", errors='raise') + df4: pd.DataFrame = pd.json_normalize(data=data1, record_path=None, meta='id') + data2: Dict[str, Any] = {'name': {'given': 'Mose', 'family': 'Regner'}} + df5: pd.DataFrame = pd.json_normalize(data=data2) diff --git a/tests/snippets/reshape/__init__.py b/tests/snippets/reshape/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/reshape/concat/__init__.py b/tests/snippets/reshape/concat/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/reshape/concat/test_typing.py b/tests/snippets/reshape/concat/test_typing.py new file mode 100644 index 0000000..02c2ad8 --- /dev/null +++ b/tests/snippets/reshape/concat/test_typing.py @@ -0,0 +1,39 @@ +# flake8: noqa: F841 +from typing import Union + +import pandas as pd + + +def test_types_concat() -> None: + s = pd.Series([0, 1, -10]) + s2 = pd.Series([7, -5, 10]) + + pd.concat([s, s2]) + pd.concat([s, s2], axis=1) + pd.concat([s, s2], keys=['first', 'second'], sort=True) + pd.concat([s, s2], keys=['first', 'second'], names=["source", "row"]) + + # Depends on the axis + rs1: Union[pd.Series, pd.DataFrame] = pd.concat({'a': s, 'b': s2}) + rs1a: Union[pd.Series, pd.DataFrame] = pd.concat({'a': s, 'b': s2}, axis=1) + rs2: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, 2: s2}) + rs2a: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, 2: s2}, axis=1) + rs3: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, None: s2}) + rs3a: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, None: s2}, axis=1) + + df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]}) + df2 = pd.DataFrame(data={'col1': [10, 20], 'col2': [30, 40]}) + + pd.concat([df, df2]) + pd.concat([df, df2], axis=1) + pd.concat([df, df2], keys=['first', 'second'], sort=True) + pd.concat([df, df2], keys=['first', 'second'], names=["source", "row"]) + + result: pd.DataFrame = pd.concat({"a": pd.DataFrame([1, 2, 3]), "b": pd.DataFrame([4, 5, 6])}, axis=1) + result2: Union[pd.DataFrame, pd.Series] = pd.concat({"a": pd.Series([1, 2, 3]), "b": pd.Series([4, 5, 6])}, axis=1) + + rdf1: pd.DataFrame = pd.concat({'a': df, 'b': df2}) + rdf2: pd.DataFrame = pd.concat({1: df, 2: df2}) + rdf3: pd.DataFrame = pd.concat({1: df, None: df2}) + + rdf4: pd.DataFrame = pd.concat(list(map(lambda x: s2, ["some_value", 3])), axis=1) diff --git a/tests/snippets/scalar/__init__.py b/tests/snippets/scalar/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/scalar/interval/__init__.py b/tests/snippets/scalar/interval/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/test_interval.py b/tests/snippets/scalar/interval/test_typing.py similarity index 100% rename from tests/snippets/test_interval.py rename to tests/snippets/scalar/interval/test_typing.py diff --git a/tests/snippets/scalar/timestamp/__init__.py b/tests/snippets/scalar/timestamp/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/test_timestamp.py b/tests/snippets/scalar/timestamp/test_typing.py similarity index 100% rename from tests/snippets/test_timestamp.py rename to tests/snippets/scalar/timestamp/test_typing.py diff --git a/tests/snippets/series/__init__.py b/tests/snippets/series/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/test_series.py b/tests/snippets/series/test_typing.py similarity index 100% rename from tests/snippets/test_series.py rename to tests/snippets/series/test_typing.py diff --git a/tests/snippets/test_pandas.py b/tests/snippets/test_pandas.py deleted file mode 100644 index 2866fe9..0000000 --- a/tests/snippets/test_pandas.py +++ /dev/null @@ -1,83 +0,0 @@ -# flake8: noqa: F841 -import tempfile -from typing import Any, Dict, List, Union - -from pandas.io.parsers import TextFileReader - -import pandas as pd - - -def test_types_to_datetime() -> None: - df = pd.DataFrame({'year': [2015, 2016], - 'month': [2, 3], - 'day': [4, 5]}) - pd.to_datetime(df) - pd.to_datetime(df, unit="s", origin="unix", infer_datetime_format=True) - pd.to_datetime(df, unit="ns", dayfirst=True, utc=None, format="%M:%D", exact=False) - pd.to_datetime([1, 2], unit="D", origin=pd.Timestamp("01/01/2000")) - pd.to_datetime([1, 2], unit="D", origin=3) - - -def test_types_concat() -> None: - s = pd.Series([0, 1, -10]) - s2 = pd.Series([7, -5, 10]) - - pd.concat([s, s2]) - pd.concat([s, s2], axis=1) - pd.concat([s, s2], keys=['first', 'second'], sort=True) - pd.concat([s, s2], keys=['first', 'second'], names=["source", "row"]) - - # Depends on the axis - rs1: Union[pd.Series, pd.DataFrame] = pd.concat({'a': s, 'b': s2}) - rs1a: Union[pd.Series, pd.DataFrame] = pd.concat({'a': s, 'b': s2}, axis=1) - rs2: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, 2: s2}) - rs2a: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, 2: s2}, axis=1) - rs3: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, None: s2}) - rs3a: Union[pd.Series, pd.DataFrame] = pd.concat({1: s, None: s2}, axis=1) - - df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]}) - df2 = pd.DataFrame(data={'col1': [10, 20], 'col2': [30, 40]}) - - pd.concat([df, df2]) - pd.concat([df, df2], axis=1) - pd.concat([df, df2], keys=['first', 'second'], sort=True) - pd.concat([df, df2], keys=['first', 'second'], names=["source", "row"]) - - result: pd.DataFrame = pd.concat({"a": pd.DataFrame([1, 2, 3]), "b": pd.DataFrame([4, 5, 6])}, axis=1) - result2: Union[pd.DataFrame, pd.Series] = pd.concat({"a": pd.Series([1, 2, 3]), "b": pd.Series([4, 5, 6])}, axis=1) - - rdf1: pd.DataFrame = pd.concat({'a': df, 'b': df2}) - rdf2: pd.DataFrame = pd.concat({1: df, 2: df2}) - rdf3: pd.DataFrame = pd.concat({1: df, None: df2}) - - rdf4: pd.DataFrame = pd.concat(list(map(lambda x: s2, ["some_value", 3])), axis=1) - - -def test_types_json_normalize() -> None: - data1: List[Dict[str, Any]] = [{'id': 1, 'name': {'first': 'Coleen', 'last': 'Volk'}}, - {'name': {'given': 'Mose', 'family': 'Regner'}}, - {'id': 2, 'name': 'Faye Raker'}] - df1: pd.DataFrame = pd.json_normalize(data=data1) - df2: pd.DataFrame = pd.json_normalize(data=data1, max_level=0, sep=";") - df3: pd.DataFrame = pd.json_normalize(data=data1, meta_prefix="id", record_prefix="name", errors='raise') - df4: pd.DataFrame = pd.json_normalize(data=data1, record_path=None, meta='id') - data2: Dict[str, Any] = {'name': {'given': 'Mose', 'family': 'Regner'}} - df5: pd.DataFrame = pd.json_normalize(data=data2) - - -def test_types_read_csv() -> None: - df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]}) - csv_df: str = df.to_csv() - - with tempfile.NamedTemporaryFile() as file: - df.to_csv(file.name) - df2: pd.DataFrame = pd.read_csv(file.name) - df3: pd.DataFrame = pd.read_csv(file.name, sep="a", squeeze=False) - df4: pd.DataFrame = pd.read_csv(file.name, header=None, prefix="b", mangle_dupe_cols=True, keep_default_na=False) - df5: pd.DataFrame = pd.read_csv(file.name, engine='python', true_values=[0, 1, 3], na_filter=False) - df6: pd.DataFrame = pd.read_csv(file.name, skiprows=lambda x: x in [0, 2], skip_blank_lines=True, dayfirst=False) - df7: pd.DataFrame = pd.read_csv(file.name, nrows=2) - tfr1: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=True, chunksize=3) - tfr2: TextFileReader = pd.read_csv(file.name, nrows=2, chunksize=1) - tfr3: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=False, chunksize=1) - tfr4: TextFileReader = pd.read_csv(file.name, nrows=2, iterator=True) diff --git a/tests/snippets/tools/__init__.py b/tests/snippets/tools/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/tools/test_to_datetime_typing.py b/tests/snippets/tools/test_to_datetime_typing.py new file mode 100644 index 0000000..c6d2b5b --- /dev/null +++ b/tests/snippets/tools/test_to_datetime_typing.py @@ -0,0 +1,14 @@ +# flake8: noqa: F841 + +import pandas as pd + + +def test_types_to_datetime() -> None: + df = pd.DataFrame({'year': [2015, 2016], + 'month': [2, 3], + 'day': [4, 5]}) + dt1: pd.datetime = pd.to_datetime(df) + dt2: pd.datetime = pd.to_datetime(df, unit="s", origin="unix", infer_datetime_format=True) + dt3: pd.datetime = pd.to_datetime(df, unit="ns", dayfirst=True, utc=None, format="%M:%D", exact=False) + dt4: pd.datetime = pd.to_datetime([1, 2], unit="D", origin=pd.Timestamp("01/01/2000")) + dt5: pd.datetime = pd.to_datetime([1, 2], unit="D", origin=3) diff --git a/tests/snippets/util/__init__.py b/tests/snippets/util/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/snippets/test_testing.py b/tests/snippets/util/test_assert_series_equal_typing.py similarity index 100% rename from tests/snippets/test_testing.py rename to tests/snippets/util/test_assert_series_equal_typing.py