MCPcopy Index your code
hub / github.com/pydata/xarray / test_find

Function test_find

xarray/tests/test_accessor_str.py:1770–1850  ·  view source on GitHub ↗
(dtype)

Source from the content-addressed store, hash-verified

1768
1769
1770def test_find(dtype) -> None:
1771 values = xr.DataArray(["ABCDEFG", "BCDEFEF", "DEFGHIJEF", "EFGHEF", "XXX"])
1772 values = values.astype(dtype)
1773
1774 result_0 = values.str.find("EF")
1775 result_1 = values.str.find("EF", side="left")
1776 expected_0 = xr.DataArray([4, 3, 1, 0, -1])
1777 expected_1 = xr.DataArray([v.find(dtype("EF")) for v in values.values])
1778 assert result_0.dtype == expected_0.dtype
1779 assert result_0.dtype == expected_1.dtype
1780 assert result_1.dtype == expected_0.dtype
1781 assert result_1.dtype == expected_1.dtype
1782 assert_equal(result_0, expected_0)
1783 assert_equal(result_0, expected_1)
1784 assert_equal(result_1, expected_0)
1785 assert_equal(result_1, expected_1)
1786
1787 result_0 = values.str.rfind("EF")
1788 result_1 = values.str.find("EF", side="right")
1789 expected_0 = xr.DataArray([4, 5, 7, 4, -1])
1790 expected_1 = xr.DataArray([v.rfind(dtype("EF")) for v in values.values])
1791 assert result_0.dtype == expected_0.dtype
1792 assert result_0.dtype == expected_1.dtype
1793 assert result_1.dtype == expected_0.dtype
1794 assert result_1.dtype == expected_1.dtype
1795 assert_equal(result_0, expected_0)
1796 assert_equal(result_0, expected_1)
1797 assert_equal(result_1, expected_0)
1798 assert_equal(result_1, expected_1)
1799
1800 result_0 = values.str.find("EF", 3)
1801 result_1 = values.str.find("EF", 3, side="left")
1802 expected_0 = xr.DataArray([4, 3, 7, 4, -1])
1803 expected_1 = xr.DataArray([v.find(dtype("EF"), 3) for v in values.values])
1804 assert result_0.dtype == expected_0.dtype
1805 assert result_0.dtype == expected_1.dtype
1806 assert result_1.dtype == expected_0.dtype
1807 assert result_1.dtype == expected_1.dtype
1808 assert_equal(result_0, expected_0)
1809 assert_equal(result_0, expected_1)
1810 assert_equal(result_1, expected_0)
1811 assert_equal(result_1, expected_1)
1812
1813 result_0 = values.str.rfind("EF", 3)
1814 result_1 = values.str.find("EF", 3, side="right")
1815 expected_0 = xr.DataArray([4, 5, 7, 4, -1])
1816 expected_1 = xr.DataArray([v.rfind(dtype("EF"), 3) for v in values.values])
1817 assert result_0.dtype == expected_0.dtype
1818 assert result_0.dtype == expected_1.dtype
1819 assert result_1.dtype == expected_0.dtype
1820 assert result_1.dtype == expected_1.dtype
1821 assert_equal(result_0, expected_0)
1822 assert_equal(result_0, expected_1)
1823 assert_equal(result_1, expected_0)
1824 assert_equal(result_1, expected_1)
1825
1826 result_0 = values.str.find("EF", 3, 6)
1827 result_1 = values.str.find("EF", 3, 6, side="left")

Callers

nothing calls this directly

Calls 5

assert_equalFunction · 0.90
findMethod · 0.80
rfindMethod · 0.80
dtypeFunction · 0.70
astypeMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…