| 1723 | |
| 1724 | @pytest.mark.parametrize("join", ["inner", "outer", "left", "right"]) |
| 1725 | def test_align(join): |
| 1726 | df1a = pd.DataFrame( |
| 1727 | {"A": np.random.randn(10), "B": np.random.randn(10)}, |
| 1728 | index=[1, 12, 5, 6, 3, 9, 10, 4, 13, 11], |
| 1729 | ) |
| 1730 | |
| 1731 | df1b = pd.DataFrame( |
| 1732 | {"A": np.random.randn(10), "B": np.random.randn(10)}, |
| 1733 | index=[0, 3, 2, 10, 5, 6, 7, 8, 12, 13], |
| 1734 | ) |
| 1735 | ddf1a = dd.from_pandas(df1a, 3) |
| 1736 | ddf1b = dd.from_pandas(df1b, 3) |
| 1737 | |
| 1738 | # DataFrame |
| 1739 | res1, res2 = ddf1a.align(ddf1b, join=join) |
| 1740 | exp1, exp2 = df1a.align(df1b, join=join) |
| 1741 | assert assert_eq(res1, exp1) |
| 1742 | assert assert_eq(res2, exp2) |
| 1743 | |
| 1744 | # Series |
| 1745 | res1, res2 = ddf1a["A"].align(ddf1b["B"], join=join) |
| 1746 | exp1, exp2 = df1a["A"].align(df1b["B"], join=join) |
| 1747 | assert assert_eq(res1, exp1) |
| 1748 | assert assert_eq(res2, exp2) |
| 1749 | |
| 1750 | # DataFrame with fill_value |
| 1751 | res1, res2 = ddf1a.align(ddf1b, join=join, fill_value=1) |
| 1752 | exp1, exp2 = df1a.align(df1b, join=join, fill_value=1) |
| 1753 | assert assert_eq(res1, exp1) |
| 1754 | assert assert_eq(res2, exp2) |
| 1755 | |
| 1756 | # Series |
| 1757 | res1, res2 = ddf1a["A"].align(ddf1b["B"], join=join, fill_value=1) |
| 1758 | exp1, exp2 = df1a["A"].align(df1b["B"], join=join, fill_value=1) |
| 1759 | assert assert_eq(res1, exp1) |
| 1760 | assert assert_eq(res2, exp2) |
| 1761 | |
| 1762 | |
| 1763 | @pytest.mark.parametrize("join", ["inner", "outer", "left", "right"]) |