()
| 706 | |
| 707 | |
| 708 | def test_load_series(): |
| 709 | pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 710 | data = pd.read_csv(TRAIN_FILE, header=None, delimiter='\t') |
| 711 | labels = pd.Series(data.iloc[:, TARGET_IDX]) |
| 712 | data.drop([TARGET_IDX], axis=1, inplace=True) |
| 713 | data = pd.Series(list(data.values)) |
| 714 | cat_features = pool.get_cat_feature_indices() |
| 715 | pool2 = Pool(data, labels, cat_features) |
| 716 | assert _have_equal_features(pool, pool2) |
| 717 | assert _check_data([int(label) for label in pool.get_label()], pool2.get_label()) |
| 718 | |
| 719 | |
| 720 | def test_pool_cat_features(): |
nothing calls this directly
no test coverage detected