| 1249 | |
| 1250 | |
| 1251 | def test_isin(): |
| 1252 | f_list = [1, 2, 3] |
| 1253 | f_series = pd.Series(f_list) |
| 1254 | f_dict = {"a": [0, 3], "b": [1, 2]} |
| 1255 | f_list2 = [1, "2"] |
| 1256 | f_list_delayed = [delayed(1), delayed(2), delayed(3)] |
| 1257 | f_list_of_lists = [[1, 2, 3], [2, 3, 4], [3, 4, 5]] |
| 1258 | |
| 1259 | # Series |
| 1260 | assert_eq(d.a.isin(f_list), full.a.isin(f_list)) |
| 1261 | assert_eq(d.a.isin(f_series), full.a.isin(f_series)) |
| 1262 | assert_eq(d.a.isin(f_list2), full.a.isin(f_list2)) |
| 1263 | assert_eq( |
| 1264 | d.a.isin(f_list_delayed), |
| 1265 | full.a.isin(f_list), |
| 1266 | ) |
| 1267 | assert_eq( |
| 1268 | d.a.isin(f_list_of_lists), |
| 1269 | full.a.isin(f_list_of_lists), |
| 1270 | ) |
| 1271 | with pytest.raises(NotImplementedError): |
| 1272 | d.a.isin(d.a) |
| 1273 | |
| 1274 | # DataFrame test |
| 1275 | assert_eq(d.isin(f_list), full.isin(f_list)) |
| 1276 | assert_eq(d.isin(f_dict), full.isin(f_dict)) |
| 1277 | assert_eq(d.isin(f_list2), full.isin(f_list2)) |
| 1278 | assert_eq( |
| 1279 | d.isin(f_list_delayed), |
| 1280 | full.isin(f_list), |
| 1281 | ) |
| 1282 | assert_eq( |
| 1283 | d.isin(f_list_of_lists), |
| 1284 | full.isin(f_list_of_lists), |
| 1285 | ) |
| 1286 | for obj in [d, f_series, full]: |
| 1287 | with pytest.raises(NotImplementedError): |
| 1288 | d.isin(obj) |
| 1289 | |
| 1290 | |
| 1291 | def test_contains_frame(): |