def test_get_data_with_mapped_label():
mapping = get_phishing_map_labels(
"Phishing:malicious , malware : malicious, spam")
data = [{
"tag": "Phishing"
}, {
"tag": "malware"
}, {
"tag": "spam"
}, {
"tag": "not"
}, {
"tag": "not"
}]
new_data, exist_labels_counter, missing_labels_counter = get_data_with_mapped_label(
data, mapping, "tag")
assert new_data[0]['tag'] == 'malicious'
assert new_data[2]['tag'] == 'spam'
assert exist_labels_counter['Phishing'] == 1
assert exist_labels_counter['malware'] == 1
assert exist_labels_counter['spam'] == 1
assert missing_labels_counter['not'] == 2
mapping = get_phishing_map_labels("phishing:malicious,spam:spam")
data = [{"tag": "Phishing"}, {"tag": "Spam"}]
new_data, exist_labels_counter, missing_labels_counter = get_data_with_mapped_label(
data, mapping, "tag")
assert new_data[0]['tag'] == 'malicious'
assert new_data[1]['tag'] == 'spam'
assert exist_labels_counter['phishing'] == 1
assert exist_labels_counter['spam'] == 1
def test_get_phishing_map_labels(mocker):
mapping = get_phishing_map_labels("Phishing:malicious , malware : malicious, spam")
assert mapping['Phishing'] == "malicious"
assert mapping['malware'] == "malicious"
assert mapping['spam'] == "spam"
mapping = get_phishing_map_labels(ALL_LABELS)
assert mapping == ALL_LABELS