96 lines
2.6 KiB
C
96 lines
2.6 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <stdarg.h>
|
|
|
|
#include "greatest.h"
|
|
#include "../src/libpostal.h"
|
|
|
|
SUITE(libpostal_expansion_tests);
|
|
|
|
static greatest_test_res test_expansion_contains(char *input, char *output, normalize_options_t options) {
|
|
size_t num_expansions;
|
|
char **expansions = expand_address(input, options, &num_expansions);
|
|
|
|
bool contains_expansion = false;
|
|
char *expansion;
|
|
for (size_t i = 0; i < num_expansions; i++) {
|
|
expansion = expansions[i];
|
|
if (strcmp(output, expansion) == 0) {
|
|
contains_expansion = true;
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
ASSERT(contains_expansion);
|
|
PASS();
|
|
}
|
|
|
|
static greatest_test_res test_expansion_contains_with_languages(char *input, char *output, normalize_options_t options, size_t num_languages, ...) {
|
|
char **languages = NULL;
|
|
|
|
size_t i;
|
|
|
|
if (num_languages > 0) {
|
|
va_list args;
|
|
|
|
va_start(args, num_languages);
|
|
languages = malloc(sizeof(char *) * num_languages);
|
|
char *lang;
|
|
|
|
for (i = 0; i < num_languages; i++) {
|
|
lang = va_arg(args, char *);
|
|
ASSERT(strlen(lang) < MAX_LANGUAGE_LEN);
|
|
languages[i] = strdup(lang);
|
|
}
|
|
|
|
va_end(args);
|
|
|
|
options.num_languages = num_languages;
|
|
options.languages = (char **)languages;
|
|
} else {
|
|
options.languages = NULL;
|
|
options.num_languages = 0;
|
|
}
|
|
|
|
CHECK_CALL(test_expansion_contains(input, output, options));
|
|
if (languages != NULL) {
|
|
for (i = 0; i < num_languages; i++) {
|
|
free(languages[i]);
|
|
}
|
|
free(languages);
|
|
}
|
|
PASS();
|
|
}
|
|
|
|
|
|
TEST test_expansions(void) {
|
|
CHECK_CALL(test_expansion_contains_with_languages("123 Main St. #2f", "123 main street number 2f", LIBPOSTAL_DEFAULT_OPTIONS, 1, "en"));
|
|
CHECK_CALL(test_expansion_contains_with_languages("Marktstrasse", "markt straße", LIBPOSTAL_DEFAULT_OPTIONS, 1, "de"));
|
|
PASS();
|
|
}
|
|
|
|
TEST test_expansions_language_classifier(void) {
|
|
CHECK_CALL(test_expansion_contains_with_languages("V XX Sett", "via 20 settembre", LIBPOSTAL_DEFAULT_OPTIONS, 0, NULL));
|
|
CHECK_CALL(test_expansion_contains_with_languages("C/ Ocho", "calle 8", LIBPOSTAL_DEFAULT_OPTIONS, 0, NULL));
|
|
PASS();
|
|
}
|
|
|
|
|
|
SUITE(libpostal_expansion_tests) {
|
|
|
|
if (!libpostal_setup() || !libpostal_setup_language_classifier()) {
|
|
printf("Could not setup libpostal\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
RUN_TEST(test_expansions);
|
|
RUN_TEST(test_expansions_language_classifier);
|
|
|
|
libpostal_teardown();
|
|
libpostal_teardown_language_classifier();
|
|
|
|
}
|
|
|