@@ -6472,18 +6472,19 @@ def test_clip(self):
6472
6472
6473
6473
def test_dataframe_clip (self ):
6474
6474
6475
- lb = - 1
6476
- ub = 1
6477
-
6478
6475
# GH #2747
6479
6476
df = DataFrame (np .random .randn (1000 ,2 ))
6480
- lb_mask = df .values <= lb
6481
- ub_mask = df .values >= ub
6482
- mask = ~ lb_mask & ~ ub_mask
6483
- clipped_df = df .clip (lb , ub )
6484
- self .assert_ ((clipped_df .values [lb_mask ] == lb ).all () == True )
6485
- self .assert_ ((clipped_df .values [ub_mask ] == ub ).all () == True )
6486
- self .assert_ ((clipped_df .values [mask ] == df .values [mask ]).all () == True )
6477
+
6478
+ for lb , ub in [(- 1 ,1 ),(1 ,- 1 )]:
6479
+ clipped_df = df .clip (lb , ub )
6480
+
6481
+ lb , ub = min (lb ,ub ), max (ub ,lb )
6482
+ lb_mask = df .values <= lb
6483
+ ub_mask = df .values >= ub
6484
+ mask = ~ lb_mask & ~ ub_mask
6485
+ self .assert_ ((clipped_df .values [lb_mask ] == lb ).all () == True )
6486
+ self .assert_ ((clipped_df .values [ub_mask ] == ub ).all () == True )
6487
+ self .assert_ ((clipped_df .values [mask ] == df .values [mask ]).all () == True )
6487
6488
6488
6489
def test_get_X_columns (self ):
6489
6490
# numeric and object columns
0 commit comments