Skip to content

Commit f2390b2

Browse files
loicdiridollouloicdiridollou
committed
GH984 PR feedback
1 parent 996298b commit f2390b2

File tree

1 file changed

+145
-0
lines changed

1 file changed

+145
-0
lines changed

‎tests/test_frame.py

Lines changed: 145 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -778,6 +778,151 @@ def test_dataframe_clip() -> None:
778778
type(None),
779779
)
780780

781+
# without lower
782+
check(assert_type(df.clip(upper=None, axis=None), pd.DataFrame), pd.DataFrame)
783+
check(assert_type(df.clip(upper=None, axis=None), pd.DataFrame), pd.DataFrame)
784+
check(assert_type(df.clip(upper=15, axis=None), pd.DataFrame), pd.DataFrame)
785+
check(
786+
assert_type(df.clip(upper=None, axis=None, inplace=True), pd.DataFrame),
787+
pd.DataFrame,
788+
)
789+
check(assert_type(df.clip(upper=15, axis=None, inplace=True), None), type(None))
790+
791+
check(assert_type(df.clip(upper=None, axis=0), pd.DataFrame), pd.DataFrame)
792+
check(assert_type(df.clip(upper=None, axis=0), pd.DataFrame), pd.DataFrame)
793+
check(assert_type(df.clip(upper=15, axis=0), pd.DataFrame), pd.DataFrame)
794+
check(assert_type(df.clip(upper=None, axis=0), pd.DataFrame), pd.DataFrame)
795+
check(
796+
assert_type(df.clip(upper=pd.Series([1, 2]), axis=0), pd.DataFrame),
797+
pd.DataFrame,
798+
)
799+
check(
800+
assert_type(df.clip(upper=None, axis="index", inplace=True), pd.DataFrame),
801+
pd.DataFrame,
802+
)
803+
check(
804+
assert_type(df.clip(upper=None, axis="index", inplace=True), pd.DataFrame),
805+
pd.DataFrame,
806+
)
807+
check(
808+
assert_type(df.clip(upper=15, axis="index", inplace=True), None),
809+
type(None),
810+
)
811+
check(
812+
assert_type(df.clip(upper=None, axis="index", inplace=True), pd.DataFrame),
813+
pd.DataFrame,
814+
)
815+
check(
816+
assert_type(df.clip(upper=pd.Series([1, 2]), axis="index", inplace=True), None),
817+
type(None),
818+
)
819+
check(assert_type(df.clip(upper=None, axis="index"), pd.DataFrame), pd.DataFrame)
820+
check(assert_type(df.clip(upper=None, axis="index"), pd.DataFrame), pd.DataFrame)
821+
check(assert_type(df.clip(upper=15, axis="index"), pd.DataFrame), pd.DataFrame)
822+
check(
823+
assert_type(
824+
df.clip(lower=pd.Series([1, 2]), upper=None, axis="index"), pd.DataFrame
825+
),
826+
pd.DataFrame,
827+
)
828+
check(
829+
assert_type(
830+
df.clip(lower=None, upper=pd.Series([1, 2]), axis="index"), pd.DataFrame
831+
),
832+
pd.DataFrame,
833+
)
834+
check(
835+
assert_type(
836+
df.clip(lower=None, upper=None, axis=0, inplace=True), pd.DataFrame
837+
),
838+
pd.DataFrame,
839+
)
840+
check(
841+
assert_type(df.clip(lower=5, upper=None, axis=0, inplace=True), None),
842+
type(None),
843+
)
844+
check(
845+
assert_type(df.clip(lower=None, upper=15, axis=0, inplace=True), None),
846+
type(None),
847+
)
848+
849+
# without upper
850+
check(
851+
assert_type(df.clip(lower=None, axis=None), pd.DataFrame),
852+
pd.DataFrame,
853+
)
854+
check(assert_type(df.clip(lower=5, axis=None), pd.DataFrame), pd.DataFrame)
855+
check(
856+
assert_type(df.clip(lower=None, axis=None, inplace=True), pd.DataFrame),
857+
pd.DataFrame,
858+
)
859+
check(
860+
assert_type(df.clip(lower=5, axis=None, inplace=True), None),
861+
type(None),
862+
)
863+
check(
864+
assert_type(df.clip(lower=None, axis=None, inplace=True), pd.DataFrame),
865+
pd.DataFrame,
866+
)
867+
868+
check(assert_type(df.clip(lower=None, axis=0), pd.DataFrame), pd.DataFrame)
869+
check(assert_type(df.clip(lower=5, axis=0), pd.DataFrame), pd.DataFrame)
870+
check(
871+
assert_type(df.clip(lower=pd.Series([1, 2]), axis=0), pd.DataFrame),
872+
pd.DataFrame,
873+
)
874+
check(
875+
assert_type(df.clip(lower=None, axis="index", inplace=True), pd.DataFrame),
876+
pd.DataFrame,
877+
)
878+
check(
879+
assert_type(df.clip(lower=5, axis="index", inplace=True), None),
880+
type(None),
881+
)
882+
check(
883+
assert_type(
884+
df.clip(lower=pd.Series([1, 2]), axis="index", inplace=True),
885+
None,
886+
),
887+
type(None),
888+
)
889+
check(
890+
assert_type(df.clip(lower=None, axis="index", inplace=True), pd.DataFrame),
891+
pd.DataFrame,
892+
)
893+
check(
894+
assert_type(df.clip(lower=None, axis="index"), pd.DataFrame),
895+
pd.DataFrame,
896+
)
897+
check(
898+
assert_type(df.clip(lower=5, axis="index"), pd.DataFrame),
899+
pd.DataFrame,
900+
)
901+
check(
902+
assert_type(df.clip(lower=None, axis="index"), pd.DataFrame),
903+
pd.DataFrame,
904+
)
905+
check(
906+
assert_type(df.clip(lower=pd.Series([1, 2]), axis="index"), pd.DataFrame),
907+
pd.DataFrame,
908+
)
909+
check(
910+
assert_type(df.clip(lower=None, axis="index"), pd.DataFrame),
911+
pd.DataFrame,
912+
)
913+
check(
914+
assert_type(df.clip(lower=None, axis=0, inplace=True), pd.DataFrame),
915+
pd.DataFrame,
916+
)
917+
check(
918+
assert_type(df.clip(lower=5, axis=0, inplace=True), None),
919+
type(None),
920+
)
921+
check(
922+
assert_type(df.clip(lower=None, axis=0, inplace=True), pd.DataFrame),
923+
pd.DataFrame,
924+
)
925+
781926

782927
def test_types_abs() -> None:
783928
df = pd.DataFrame(data={"col1": [-5, 1], "col2": [3, -14]})

0 commit comments

Comments
 (0)