def test_get_k_neighbors_negative(train_df):
test_df = train_df
test_df.loc[len(train_df)] = [
len(train_df),
"test",
"defect,analysis,coffee",
8,
pd.to_datetime("08-01-2020", utc=True, dayfirst=True),
1,
0,
10,
"Duplicated",
"Low",
]
with pytest.raises(SmallNumberRepresentatives):
get_k_neighbors(test_df["Description_tr"])
def test_get_k_neighbors_positive_2(train_df):
len_df = len(train_df)
test_df = train_df
for i in range(len_df, len_df + 4):
test_df.loc[i] = [
i,
"test",
"defect,analysis,coffee",
8,
pd.to_datetime("08-01-2020", utc=True, dayfirst=True),
1,
0,
10,
"Duplicated",
"Low",
]
assert get_k_neighbors(test_df["Description_tr"]) == 2
def test_get_k_neighbors_positive(train_df):
assert get_k_neighbors(train_df["Description_tr"]) == 2