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