| 1265 | |
| 1266 | |
| 1267 | def test_isin(): |
| 1268 | f_list = [1, 2, 3] |
| 1269 | f_series = pd.Series(f_list) |
| 1270 | f_dict = {"a": [0, 3], "b": [1, 2]} |
| 1271 | f_list2 = [1, "2"] |
| 1272 | f_list_delayed = [delayed(1), delayed(2), delayed(3)] |
| 1273 | f_list_of_lists = [[1, 2, 3], [2, 3, 4], [3, 4, 5]] |
| 1274 | |
| 1275 | # Series |
| 1276 | assert_eq(d.a.isin(f_list), full.a.isin(f_list)) |
| 1277 | assert_eq(d.a.isin(f_series), full.a.isin(f_series)) |
| 1278 | assert_eq(d.a.isin(f_list2), full.a.isin(f_list2)) |
| 1279 | assert_eq( |
| 1280 | d.a.isin(f_list_delayed), |
| 1281 | full.a.isin(f_list), |
| 1282 | ) |
| 1283 | assert_eq( |
| 1284 | d.a.isin(f_list_of_lists), |
| 1285 | full.a.isin(f_list_of_lists), |
| 1286 | ) |
| 1287 | with pytest.raises(NotImplementedError): |
| 1288 | d.a.isin(d.a) |
| 1289 | |
| 1290 | # DataFrame test |
| 1291 | assert_eq(d.isin(f_list), full.isin(f_list)) |
| 1292 | assert_eq(d.isin(f_dict), full.isin(f_dict)) |
| 1293 | assert_eq(d.isin(f_list2), full.isin(f_list2)) |
| 1294 | assert_eq( |
| 1295 | d.isin(f_list_delayed), |
| 1296 | full.isin(f_list), |
| 1297 | ) |
| 1298 | assert_eq( |
| 1299 | d.isin(f_list_of_lists), |
| 1300 | full.isin(f_list_of_lists), |
| 1301 | ) |
| 1302 | for obj in [d, f_series, full]: |
| 1303 | with pytest.raises(NotImplementedError): |
| 1304 | d.isin(obj) |
| 1305 | |
| 1306 | |
| 1307 | def test_contains_frame(): |