| 723 | |
| 724 | @pytest.mark.parametrize("lower, upper", [(2, 5), (2.5, 3.5)]) |
| 725 | def test_clip(lower, upper): |
| 726 | df = pd.DataFrame( |
| 727 | {"a": [1, 2, 3, 4, 5, 6, 7, 8, 9], "b": [3, 5, 2, 5, 7, 2, 4, 2, 4]} |
| 728 | ) |
| 729 | ddf = dd.from_pandas(df, 3) |
| 730 | |
| 731 | s = pd.Series([1, 2, 3, 4, 5, 6, 7, 8, 9]) |
| 732 | ds = dd.from_pandas(s, 3) |
| 733 | |
| 734 | assert_eq(ddf.clip(lower=lower, upper=upper), df.clip(lower=lower, upper=upper)) |
| 735 | assert_eq(ddf.clip(lower=lower), df.clip(lower=lower)) |
| 736 | assert_eq(ddf.clip(upper=upper), df.clip(upper=upper)) |
| 737 | |
| 738 | assert_eq(ds.clip(lower=lower, upper=upper), s.clip(lower=lower, upper=upper)) |
| 739 | assert_eq(ds.clip(lower=lower), s.clip(lower=lower)) |
| 740 | assert_eq(ds.clip(upper=upper), s.clip(upper=upper)) |
| 741 | |
| 742 | |
| 743 | def test_clip_axis_0(): |