| 739 | |
| 740 | @pytest.mark.parametrize("lower, upper", [(2, 5), (2.5, 3.5)]) |
| 741 | def test_clip(lower, upper): |
| 742 | df = pd.DataFrame( |
| 743 | {"a": [1, 2, 3, 4, 5, 6, 7, 8, 9], "b": [3, 5, 2, 5, 7, 2, 4, 2, 4]} |
| 744 | ) |
| 745 | ddf = dd.from_pandas(df, 3) |
| 746 | |
| 747 | s = pd.Series([1, 2, 3, 4, 5, 6, 7, 8, 9]) |
| 748 | ds = dd.from_pandas(s, 3) |
| 749 | |
| 750 | assert_eq(ddf.clip(lower=lower, upper=upper), df.clip(lower=lower, upper=upper)) |
| 751 | assert_eq(ddf.clip(lower=lower), df.clip(lower=lower)) |
| 752 | assert_eq(ddf.clip(upper=upper), df.clip(upper=upper)) |
| 753 | |
| 754 | assert_eq(ds.clip(lower=lower, upper=upper), s.clip(lower=lower, upper=upper)) |
| 755 | assert_eq(ds.clip(lower=lower), s.clip(lower=lower)) |
| 756 | assert_eq(ds.clip(upper=upper), s.clip(upper=upper)) |
| 757 | |
| 758 | |
| 759 | def test_clip_axis_0(): |