/* PSPP - a program for statistical analysis.
   Copyright (C) 2006, 2010 Free Software Foundation, Inc.

   This program is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>. */

/* This is a test program for the ll_* routines defined in
   ll.c.  This test program aims to be as comprehensive as
   possible.  "gcov -b" should report 100% coverage of lines and
   branches in the ll_* routines.  "valgrind --leak-check=yes
   --show-reachable=yes" should give a clean report.

   This test program depends only on ll.c and the standard C
   library.

   See llx-test.c for a similar program for the llx_*
   routines. */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <libpspp/ll.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

/* Support preliminaries. */
#if __GNUC__ >= 2 && !defined UNUSED
#define UNUSED __attribute__ ((unused))
#else
#define UNUSED
#endif

/* Exit with a failure code.
   (Place a breakpoint on this function while debugging.) */
static void
check_die (void)
{
  exit (EXIT_FAILURE);
}

/* If OK is not true, prints a message about failure on the
   current source file and the given LINE and terminates. */
static void
check_func (bool ok, int line)
{
  if (!ok)
    {
      fprintf (stderr, "%s:%d: check failed\n", __FILE__, line);
      check_die ();
    }
}

/* Verifies that EXPR evaluates to true.
   If not, prints a message citing the calling line number and
   terminates. */
#define check(EXPR) check_func ((EXPR), __LINE__)

/* Prints a message about memory exhaustion and exits with a
   failure code. */
static void
xalloc_die (void)
{
  printf ("virtual memory exhausted\n");
  exit (EXIT_FAILURE);
}

/* Allocates and returns N bytes of memory. */
static void *
xmalloc (size_t n)
{
  if (n != 0)
    {
      void *p = malloc (n);
      if (p == NULL)
        xalloc_die ();

      return p;
    }
  else
    return NULL;
}

/* Allocates and returns N * M bytes of memory. */
static void *
xnmalloc (size_t n, size_t m)
{
  if ((size_t) -1 / m <= n)
    xalloc_die ();
  return xmalloc (n * m);
}

/* List type and support routines. */

/* Test data element. */
struct element
  {
    struct ll ll;               /* Embedded list element. */
    int x;                      /* Primary value. */
    int y;                      /* Secondary value. */
  };

static int aux_data;

/* Returns the `struct element' that LL is embedded within. */
static struct element *
ll_to_element (const struct ll *ll)
{
  return ll_data (ll, struct element, ll);
}

/* Prints the elements in LIST. */
static void UNUSED
print_list (struct ll_list *list)
{
  struct ll *x;

  printf ("list:");
  for (x = ll_head (list); x != ll_null (list); x = ll_next (x))
    {
      struct element *e = ll_to_element (x);
      printf (" %d", e->x);
    }
  printf ("\n");
}

/* Prints the value returned by PREDICATE given auxiliary data
   AUX for each element in LIST. */
static void UNUSED
print_pred (struct ll_list *list,
            ll_predicate_func *predicate, void *aux UNUSED)
{
  struct ll *x;

  printf ("pred:");
  for (x = ll_head (list); x != ll_null (list); x = ll_next (x))
    printf (" %d", predicate (x, aux));
  printf ("\n");
}

/* Prints the N numbers in VALUES. */
static void UNUSED
print_array (int values[], size_t n)
{
  size_t i;

  printf ("arry:");
  for (i = 0; i < n; i++)
    printf (" %d", values[i]);
  printf ("\n");
}

/* Compares the `x' values in A and B and returns a strcmp-type
   return value.  Verifies that AUX points to aux_data. */
static int
compare_elements (const struct ll *a_, const struct ll *b_, void *aux)
{
  const struct element *a = ll_to_element (a_);
  const struct element *b = ll_to_element (b_);

  check (aux == &aux_data);
  return a->x < b->x ? -1 : a->x > b->x;
}

/* Compares the `x' and `y' values in A and B and returns a
   strcmp-type return value.  Verifies that AUX points to
   aux_data. */
static int
compare_elements_x_y (const struct ll *a_, const struct ll *b_, void *aux)
{
  const struct element *a = ll_to_element (a_);
  const struct element *b = ll_to_element (b_);

  check (aux == &aux_data);
  if (a->x != b->x)
    return a->x < b->x ? -1 : 1;
  else if (a->y != b->y)
    return a->y < b->y ? -1 : 1;
  else
    return 0;
}

/* Compares the `y' values in A and B and returns a strcmp-type
   return value.  Verifies that AUX points to aux_data. */
static int
compare_elements_y (const struct ll *a_, const struct ll *b_, void *aux)
{
  const struct element *a = ll_to_element (a_);
  const struct element *b = ll_to_element (b_);

  check (aux == &aux_data);
  return a->y < b->y ? -1 : a->y > b->y;
}

/* Returns true if the bit in *PATTERN indicated by `x in
   *ELEMENT is set, false otherwise. */
static bool
pattern_pred (const struct ll *element_, void *pattern_)
{
  const struct element *element = ll_to_element (element_);
  unsigned int *pattern = pattern_;

  return (*pattern & (1u << element->x)) != 0;
}

/* Allocates N elements in *ELEMS.
   Adds the elements to LIST, if it is nonnull.
   Puts pointers to the elements' list elements in *ELEMP,
   followed by a pointer to the list null element, if ELEMP is
   nonnull.
   Allocates space for N values in *VALUES, if VALUES is
   nonnull. */
static void
allocate_elements (size_t n,
                   struct ll_list *list,
                   struct element ***elems,
                   struct ll ***elemp,
                   int **values)
{
  size_t i;

  if (list != NULL)
    ll_init (list);

  *elems = xnmalloc (n, sizeof **elems);
  for (i = 0; i < n; i++)
    {
      (*elems)[i] = xmalloc (sizeof ***elems);
      if (list != NULL)
        ll_push_tail (list, &(*elems)[i]->ll);
    }

  if (elemp != NULL)
    {
      *elemp = xnmalloc (n + 1, sizeof *elemp);
      for (i = 0; i < n; i++)
        (*elemp)[i] = &(*elems)[i]->ll;
      (*elemp)[n] = ll_null (list);
    }

  if (values != NULL)
    *values = xnmalloc (n, sizeof *values);
}

/* Copies the N values of `x' from LIST into VALUES[]. */
static void
extract_values (struct ll_list *list, int values[], size_t n)
{
  struct ll *x;

  check (ll_count (list) == n);
  for (x = ll_head (list); x != ll_null (list); x = ll_next (x))
    {
      struct element *e = ll_to_element (x);
      *values++ = e->x;
    }
}

/* As allocate_elements, but sets ascending values, starting
   from 0, in `x' values in *ELEMS and in *VALUES (if
   nonnull). */
static void
allocate_ascending (size_t n,
                    struct ll_list *list,
                    struct element ***elems,
                    struct ll ***elemp,
                    int **values)
{
  size_t i;

  allocate_elements (n, list, elems, elemp, values);

  for (i = 0; i < n; i++)
    (*elems)[i]->x = i;
  if (values != NULL)
    extract_values (list, *values, n);
}

/* As allocate_elements, but sets binary values extracted from
   successive bits in PATTERN in `x' values in *ELEMS and in
   *VALUES (if nonnull). */
static void
allocate_pattern (size_t n,
                  int pattern,
                  struct ll_list *list,
                  struct element ***elems,
                  struct ll ***elemp,
                  int **values)
{
  size_t i;

  allocate_elements (n, list, elems, elemp, values);

  for (i = 0; i < n; i++)
    (*elems)[i]->x = (pattern & (1 << i)) != 0;
  if (values != NULL)
    extract_values (list, *values, n);
}

/* Randomly shuffles the N elements in ARRAY, each of which is
   SIZE bytes in size. */
static void
random_shuffle (void *array_, size_t n, size_t size)
{
  char *array = array_;
  char *tmp = xmalloc (size);
  size_t i;

  for (i = 0; i < n; i++)
    {
      size_t j = rand () % (n - i) + i;
      if (i != j)
        {
          memcpy (tmp, array + j * size, size);
          memcpy (array + j * size, array + i * size, size);
          memcpy (array + i * size, tmp, size);
        }
    }

  free (tmp);
}

/* As allocate_ascending, but orders the values randomly. */
static void
allocate_random (size_t n,
                 struct ll_list *list,
                 struct element ***elems,
                 struct ll ***elemp,
                 int **values)
{
  size_t i;

  allocate_elements (n, list, elems, elemp, values);

  for (i = 0; i < n; i++)
    (*elems)[i]->x = i;
  random_shuffle (*elems, n, sizeof **elems);
  if (values != NULL)
    extract_values (list, *values, n);
}

/* Frees the N elements of ELEMS, ELEMP, and VALUES. */
static void
free_elements (size_t n,
               struct element **elems,
               struct ll **elemp,
               int *values)
{
  size_t i;

  for (i = 0; i < n; i++)
    free (elems[i]);
  free (elems);
  free (elemp);
  free (values);
}

/* Compares A and B and returns a strcmp-type return value. */
static int
compare_ints (const void *a_, const void *b_, void *aux UNUSED)
{
  const int *a = a_;
  const int *b = b_;

  return *a < *b ? -1 : *a > *b;
}

/* Compares A and B and returns a strcmp-type return value. */
static int
compare_ints_noaux (const void *a_, const void *b_)
{
  const int *a = a_;
  const int *b = b_;

  return *a < *b ? -1 : *a > *b;
}

/* Checks that LIST contains the N values in ELEMENTS. */
static void
check_list_contents (struct ll_list *list, int elements[], size_t n)
{
  struct ll *ll;
  size_t i;

  check ((n == 0) == ll_is_empty (list));

  /* Iterate in forward order. */
  for (ll = ll_head (list), i = 0; i < n; ll = ll_next (ll), i++)
    {
      struct element *e = ll_to_element (ll);
      check (elements[i] == e->x);
      check (ll != ll_null (list));
    }
  check (ll == ll_null (list));

  /* Iterate in reverse order. */
  for (ll = ll_tail (list), i = 0; i < n; ll = ll_prev (ll), i++)
    {
      struct element *e = ll_to_element (ll);
      check (elements[n - i - 1] == e->x);
      check (ll != ll_null (list));
    }
  check (ll == ll_null (list));

  check (ll_count (list) == n);
}

/* Lexicographically compares ARRAY1, which contains COUNT1
   elements of SIZE bytes each, to ARRAY2, which contains COUNT2
   elements of SIZE bytes, according to COMPARE.  Returns a
   strcmp-type result.  AUX is passed to COMPARE as auxiliary
   data. */
static int
lexicographical_compare_3way (const void *array1, size_t count1,
                              const void *array2, size_t count2,
                              size_t size,
                              int (*compare) (const void *, const void *,
                                              void *aux),
                              void *aux)
{
  const char *first1 = array1;
  const char *first2 = array2;
  size_t min_count = count1 < count2 ? count1 : count2;

  while (min_count > 0)
    {
      int cmp = compare (first1, first2, aux);
      if (cmp != 0)
        return cmp;

      first1 += size;
      first2 += size;
      min_count--;
    }

  return count1 < count2 ? -1 : count1 > count2;
}

/* Tests. */

/* Tests list push and pop operations. */
static void
test_push_pop (void)
{
  const int max_elems = 1024;

  struct ll_list list;
  struct element **elems;
  int *values;

  int i;

  allocate_elements (max_elems, NULL, &elems, NULL, &values);

  /* Push on tail. */
  ll_init (&list);
  check_list_contents (&list, NULL, 0);
  for (i = 0; i < max_elems; i++)
    {
      values[i] = elems[i]->x = i;
      ll_push_tail (&list, &elems[i]->ll);
      check_list_contents (&list, values, i + 1);
    }

  /* Remove from tail. */
  for (i = 0; i < max_elems; i++)
    {
      struct element *e = ll_to_element (ll_pop_tail (&list));
      check (e->x == max_elems - i - 1);
      check_list_contents (&list, values, max_elems - i - 1);
    }

  /* Push at start. */
  check_list_contents (&list, NULL, 0);
  for (i = 0; i < max_elems; i++)
    {
      values[max_elems - i - 1] = elems[i]->x = max_elems - i - 1;
      ll_push_head (&list, &elems[i]->ll);
      check_list_contents (&list, &values[max_elems - i - 1], i + 1);
    }

  /* Remove from start. */
  for (i = 0; i < max_elems; i++)
    {
      struct element *e = ll_to_element (ll_pop_head (&list));
      check (e->x == (int) i);
      check_list_contents (&list, &values[i + 1], max_elems - i - 1);
    }

  free_elements (max_elems, elems, NULL, values);
}

/* Tests insertion and removal at arbitrary positions. */
static void
test_insert_remove (void)
{
  const int max_elems = 16;
  int n;

  for (n = 0; n < max_elems; n++)
    {
      struct element **elems;
      struct ll **elemp;
      int *values = xnmalloc (n + 1, sizeof *values);

      struct ll_list list;
      struct element extra;
      int pos;

      allocate_ascending (n, &list, &elems, &elemp, NULL);
      extra.x = -1;
      for (pos = 0; pos <= n; pos++)
        {
          int i, j;

          ll_insert (elemp[pos], &extra.ll);

          j = 0;
          for (i = 0; i < pos; i++)
            values[j++] = i;
          values[j++] = -1;
          for (; i < n; i++)
            values[j++] = i;
          check_list_contents (&list, values, n + 1);

          ll_remove (&extra.ll);
        }
      check_list_contents (&list, values, n);

      free_elements (n, elems, elemp, values);
    }
}

/* Tests swapping individual elements. */
static void
test_swap (void)
{
  const int max_elems = 8;
  int n;

  for (n = 0; n <= max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      int *values;

      int i, j, k;

      allocate_ascending (n, &list, &elems, NULL, &values);
      check_list_contents (&list, values, n);

      for (i = 0; i < n; i++)
        for (j = 0; j < n; j++)
          for (k = 0; k < 2; k++)
            {
              int t;

              ll_swap (&elems[i]->ll, &elems[j]->ll);
              t = values[i];
              values[i] = values[j];
              values[j] = t;
              check_list_contents (&list, values, n);
            }

      free_elements (n, elems, NULL, values);
    }
}

/* Tests swapping ranges of list elements. */
static void
test_swap_range (void)
{
  const int max_elems = 8;
  int n, a0, a1, b0, b1, r;

  for (n = 0; n <= max_elems; n++)
    for (a0 = 0; a0 <= n; a0++)
      for (a1 = a0; a1 <= n; a1++)
        for (b0 = a1; b0 <= n; b0++)
          for (b1 = b0; b1 <= n; b1++)
            for (r = 0; r < 2; r++)
              {
                struct ll_list list;
                struct element **elems;
                struct ll **elemp;
                int *values;

                int i, j;

                allocate_ascending (n, &list, &elems, &elemp, &values);
                check_list_contents (&list, values, n);

                j = 0;
                for (i = 0; i < a0; i++)
                  values[j++] = i;
                for (i = b0; i < b1; i++)
                  values[j++] = i;
                for (i = a1; i < b0; i++)
                  values[j++] = i;
                for (i = a0; i < a1; i++)
                  values[j++] = i;
                for (i = b1; i < n; i++)
                  values[j++] = i;
                check (j == n);

                if (r == 0)
                  ll_swap_range (elemp[a0], elemp[a1], elemp[b0], elemp[b1]);
                else
                  ll_swap_range (elemp[b0], elemp[b1], elemp[a0], elemp[a1]);
                check_list_contents (&list, values, n);

                free_elements (n, elems, elemp, values);
              }
}

/* Tests removing ranges of list elements. */
static void
test_remove_range (void)
{
  const int max_elems = 8;

  int n, r0, r1;

  for (n = 0; n <= max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        {
          struct ll_list list;
          struct element **elems;
          struct ll **elemp;
          int *values;

          int i, j;

          allocate_ascending (n, &list, &elems, &elemp, &values);
          check_list_contents (&list, values, n);

          j = 0;
          for (i = 0; i < r0; i++)
            values[j++] = i;
          for (i = r1; i < n; i++)
            values[j++] = i;

          ll_remove_range (elemp[r0], elemp[r1]);
          check_list_contents (&list, values, j);

          free_elements (n, elems, elemp, values);
        }
}

/* Tests ll_remove_equal. */
static void
test_remove_equal (void)
{
  const int max_elems = 8;

  int n, r0, r1, eq_pat;

  for (n = 0; n <= max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        for (eq_pat = 0; eq_pat <= 1 << n; eq_pat++)
          {
            struct ll_list list;
            struct element **elems;
            struct ll **elemp;
            int *values;

            struct element to_remove;
            int remaining;
            int i;

            allocate_elements (n, &list, &elems, &elemp, &values);

            remaining = 0;
            for (i = 0; i < n; i++)
              {
                int x = eq_pat & (1 << i) ? -1 : i;
                bool delete = x == -1 && r0 <= i && i < r1;
                elems[i]->x = x;
                if (!delete)
                  values[remaining++] = x;
              }

            to_remove.x = -1;
            check ((int) ll_remove_equal (elemp[r0], elemp[r1], &to_remove.ll,
                                          compare_elements, &aux_data)
                   == n - remaining);
            check_list_contents (&list, values, remaining);

            free_elements (n, elems, elemp, values);
          }
}

/* Tests ll_remove_if. */
static void
test_remove_if (void)
{
  const int max_elems = 8;

  int n, r0, r1, pattern;

  for (n = 0; n <= max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        for (pattern = 0; pattern <= 1 << n; pattern++)
          {
            struct ll_list list;
            struct element **elems;
            struct ll **elemp;
            int *values;

            int remaining;
            int i;

            allocate_elements (n, &list, &elems, &elemp, &values);

            remaining = 0;
            for (i = 0; i < n; i++)
              {
                bool delete = (pattern & (1 << i)) && r0 <= i && i < r1;
                elems[i]->x = i;
                if (!delete)
                  values[remaining++] = i;
              }

            check ((int) ll_remove_if (elemp[r0], elemp[r1],
                                       pattern_pred, &pattern)
                   == n - remaining);
            check_list_contents (&list, values, remaining);

            free_elements (n, elems, elemp, values);
          }
}

/* Tests ll_moved. */
static void
test_moved (void)
{
  const int max_elems = 8;

  int n;

  for (n = 0; n <= max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      struct element **new_elems;
      int *values;

      int i;

      allocate_ascending (n, &list, &elems, NULL, &values);
      allocate_elements (n, NULL, &new_elems, NULL, NULL);
      check_list_contents (&list, values, n);

      for (i = 0; i < n; i++)
        {
          *new_elems[i] = *elems[i];
          ll_moved (&new_elems[i]->ll);
          check_list_contents (&list, values, n);
        }

      free_elements (n, elems, NULL, values);
      free_elements (n, new_elems, NULL, NULL);
    }
}

/* Tests, via HELPER, a function that looks at list elements
   equal to some specified element. */
static void
test_examine_equal_range (void (*helper) (int r0, int r1, int eq_pat,
                                          struct ll *to_find,
                                          struct ll **elemp))
{
  const int max_elems = 8;

  int n, r0, r1, eq_pat;

  for (n = 0; n <= max_elems; n++)
    for (eq_pat = 0; eq_pat <= 1 << n; eq_pat++)
      {
        struct ll_list list;
        struct element **elems;
        struct ll **elemp;
        int *values;

        struct element to_find;

        int i;

        allocate_ascending (n, &list, &elems, &elemp, &values);

        for (i = 0; i < n; i++)
          if (eq_pat & (1 << i))
            values[i] = elems[i]->x = -1;

        to_find.x = -1;
        for (r0 = 0; r0 <= n; r0++)
          for (r1 = r0; r1 <= n; r1++)
            helper (r0, r1, eq_pat, &to_find.ll, elemp);

        check_list_contents (&list, values, n);

        free_elements (n, elems, elemp, values);
      }
}

/* Tests, via HELPER, a function that looks at list elements for
   which a given predicate returns true. */
static void
test_examine_if_range (void (*helper) (int r0, int r1, int eq_pat,
                                       struct ll **elemp))
{
  const int max_elems = 8;

  int n, r0, r1, eq_pat;

  for (n = 0; n <= max_elems; n++)
    for (eq_pat = 0; eq_pat <= 1 << n; eq_pat++)
      {
        struct ll_list list;
        struct element **elems;
        struct ll **elemp;
        int *values;

        allocate_ascending (n, &list, &elems, &elemp, &values);

        for (r0 = 0; r0 <= n; r0++)
          for (r1 = r0; r1 <= n; r1++)
            helper (r0, r1, eq_pat, elemp);

        check_list_contents (&list, values, n);

        free_elements (n, elems, elemp, values);
      }
}

/* Helper function for testing ll_find_equal. */
static void
test_find_equal_helper (int r0, int r1, int eq_pat,
                        struct ll *to_find, struct ll **elemp)
{
  struct ll *match;
  int i;

  match = ll_find_equal (elemp[r0], elemp[r1], to_find,
                         compare_elements, &aux_data);
  for (i = r0; i < r1; i++)
    if (eq_pat & (1 << i))
      break;

  check (match == elemp[i]);
}

/* Tests ll_find_equal. */
static void
test_find_equal (void)
{
  test_examine_equal_range (test_find_equal_helper);
}

/* Helper function for testing ll_find_if. */
static void
test_find_if_helper (int r0, int r1, int eq_pat, struct ll **elemp)
{
  struct ll *match = ll_find_if (elemp[r0], elemp[r1],
                                 pattern_pred, &eq_pat);
  int i;

  for (i = r0; i < r1; i++)
    if (eq_pat & (1 << i))
      break;

  check (match == elemp[i]);
}

/* Tests ll_find_if. */
static void
test_find_if (void)
{
  test_examine_if_range (test_find_if_helper);
}

/* Tests ll_find_adjacent_equal. */
static void
test_find_adjacent_equal (void)
{
  const int max_elems = 8;

  int n, eq_pat;

  for (n = 0; n <= max_elems; n++)
    for (eq_pat = 0; eq_pat <= 1 << n; eq_pat++)
      {
        struct ll_list list;
        struct element **elems;
        struct ll **elemp;
        int *values;
        int match;

        int i;

        allocate_ascending (n, &list, &elems, &elemp, &values);

        match = -1;
        for (i = 0; i < n - 1; i++)
          {
            elems[i]->y = i;
            if (eq_pat & (1 << i))
              {
                values[i] = elems[i]->x = match;
                values[i + 1] = elems[i + 1]->x = match;
              }
            else
              match--;
          }

        for (i = 0; i <= n; i++)
          {
            struct ll *ll1 = ll_find_adjacent_equal (elemp[i], ll_null (&list),
                                                     compare_elements,
                                                     &aux_data);
            struct ll *ll2;
            int j;

            ll2 = ll_null (&list);
            for (j = i; j < n - 1; j++)
              if (eq_pat & (1 << j))
                {
                  ll2 = elemp[j];
                  break;
                }
            check (ll1 == ll2);
          }
        check_list_contents (&list, values, n);

        free_elements (n, elems, elemp, values);
      }
}

/* Helper function for testing ll_count_range. */
static void
test_count_range_helper (int r0, int r1, int eq_pat UNUSED, struct ll **elemp)
{
  check ((int) ll_count_range (elemp[r0], elemp[r1]) == r1 - r0);
}

/* Tests ll_count_range. */
static void
test_count_range (void)
{
  test_examine_if_range (test_count_range_helper);
}

/* Helper function for testing ll_count_equal. */
static void
test_count_equal_helper (int r0, int r1, int eq_pat,
                         struct ll *to_find, struct ll **elemp)
{
  int count1, count2;
  int i;

  count1 = ll_count_equal (elemp[r0], elemp[r1], to_find,
                           compare_elements, &aux_data);
  count2 = 0;
  for (i = r0; i < r1; i++)
    if (eq_pat & (1 << i))
      count2++;

  check (count1 == count2);
}

/* Tests ll_count_equal. */
static void
test_count_equal (void)
{
  test_examine_equal_range (test_count_equal_helper);
}

/* Helper function for testing ll_count_if. */
static void
test_count_if_helper (int r0, int r1, int eq_pat, struct ll **elemp)
{
  int count1;
  int count2;
  int i;

  count1 = ll_count_if (elemp[r0], elemp[r1], pattern_pred, &eq_pat);

  count2 = 0;
  for (i = r0; i < r1; i++)
    if (eq_pat & (1 << i))
      count2++;

  check (count1 == count2);
}

/* Tests ll_count_if. */
static void
test_count_if (void)
{
  test_examine_if_range (test_count_if_helper);
}

/* Returns N!. */
static unsigned int
factorial (unsigned int n)
{
  unsigned int value = 1;
  while (n > 1)
    value *= n--;
  return value;
}

/* Returns the number of permutations of the N values in
   VALUES.  If VALUES contains duplicates, they must be
   adjacent. */
static unsigned int
expected_perms (int *values, size_t n)
{
  size_t i, j;
  unsigned int n_perms;

  n_perms = factorial (n);
  for (i = 0; i < n; i = j)
    {
      for (j = i + 1; j < n; j++)
        if (values[i] != values[j])
          break;
      n_perms /= factorial (j - i);
    }
  return n_perms;
}

/* Tests ll_min and ll_max. */
static void
test_min_max (void)
{
  const int max_elems = 6;
  int n;

  for (n = 0; n <= max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      struct ll **elemp;
      int *values;
      int *new_values = xnmalloc (n, sizeof *values);

      size_t n_perms;

      allocate_ascending (n, &list, &elems, &elemp, &values);

      n_perms = 1;
      while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                  compare_elements, &aux_data))
        {
          int r0, r1;
          struct ll *x;
          int i;

          for (i = 0, x = ll_head (&list); x != ll_null (&list);
               x = ll_next (x), i++)
            {
              struct element *e = ll_to_element (x);
              elemp[i] = x;
              new_values[i] = e->x;
            }
          for (r0 = 0; r0 <= n; r0++)
            for (r1 = r0; r1 <= n; r1++)
              {
                struct ll *min = ll_min (elemp[r0], elemp[r1],
                                         compare_elements, &aux_data);
                struct ll *max = ll_max (elemp[r0], elemp[r1],
                                         compare_elements, &aux_data);
                if (r0 == r1)
                  {
                    check (min == elemp[r1]);
                    check (max == elemp[r1]);
                  }
                else
                  {
                    int min_int, max_int;
                    int i;

                    min_int = max_int = new_values[r0];
                    for (i = r0; i < r1; i++)
                      {
                        int value = new_values[i];
                        if (value < min_int)
                          min_int = value;
                        if (value > max_int)
                          max_int = value;
                      }
                    check (min != elemp[r1]
                           && ll_to_element (min)->x == min_int);
                    check (max != elemp[r1]
                           && ll_to_element (max)->x == max_int);
                  }
              }
          n_perms++;
        }
      check (n_perms == factorial (n));
      check_list_contents (&list, values, n);

      free_elements (n, elems, elemp, values);
      free (new_values);
    }
}

/* Tests ll_lexicographical_compare_3way. */
static void
test_lexicographical_compare_3way (void)
{
  const int max_elems = 4;

  int n_a, pat_a, n_b, pat_b;

  for (n_a = 0; n_a <= max_elems; n_a++)
    for (pat_a = 0; pat_a <= 1 << n_a; pat_a++)
      for (n_b = 0; n_b <= max_elems; n_b++)
        for (pat_b = 0; pat_b <= 1 << n_b; pat_b++)
          {
            struct ll_list list_a, list_b;
            struct element **elems_a, **elems_b;
            struct ll **elemp_a, **elemp_b;
            int *values_a, *values_b;

            int a0, a1, b0, b1;

            allocate_pattern (n_a, pat_a,
                              &list_a, &elems_a, &elemp_a, &values_a);
            allocate_pattern (n_b, pat_b,
                              &list_b, &elems_b, &elemp_b, &values_b);

            for (a0 = 0; a0 <= n_a; a0++)
              for (a1 = a0; a1 <= n_a; a1++)
                for (b0 = 0; b0 <= n_b; b0++)
                  for (b1 = b0; b1 <= n_b; b1++)
                    {
                      int a_ordering = lexicographical_compare_3way (
                        values_a + a0, a1 - a0,
                        values_b + b0, b1 - b0,
                        sizeof *values_a,
                        compare_ints, NULL);

                      int b_ordering = ll_lexicographical_compare_3way (
                        elemp_a[a0], elemp_a[a1],
                        elemp_b[b0], elemp_b[b1],
                        compare_elements, &aux_data);

                      check (a_ordering == b_ordering);
                    }

            free_elements (n_a, elems_a, elemp_a, values_a);
            free_elements (n_b, elems_b, elemp_b, values_b);
          }
}

/* Appends the `x' value in element E to the array pointed to by
   NEXT_OUTPUT, and advances NEXT_OUTPUT to the next position. */
static void
apply_func (struct ll *e_, void *next_output_)
{
  struct element *e = ll_to_element (e_);
  int **next_output = next_output_;

  *(*next_output)++ = e->x;
}

/* Tests ll_apply. */
static void
test_apply (void)
{
  const int max_elems = 8;

  int n, r0, r1;

  for (n = 0; n <= max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        {
          struct ll_list list;
          struct element **elems;
          struct ll **elemp;
          int *values;

          int *output;
          int *next_output;

          int i;

          allocate_ascending (n, &list, &elems, &elemp, &values);
          check_list_contents (&list, values, n);

          output = next_output = xnmalloc (n, sizeof *output);
          ll_apply (elemp[r0], elemp[r1], apply_func, &next_output);
          check_list_contents (&list, values, n);

          check (r1 - r0 == next_output - output);
          for (i = 0; i < r1 - r0; i++)
            check (output[i] == r0 + i);

          free_elements (n, elems, elemp, values);
          free (output);
        }
}

/* Tests ll_reverse. */
static void
test_reverse (void)
{
  const int max_elems = 8;

  int n, r0, r1;

  for (n = 0; n <= max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        {
          struct ll_list list;
          struct element **elems;
          struct ll **elemp;
          int *values;

          int i, j;

          allocate_ascending (n, &list, &elems, &elemp, &values);
          check_list_contents (&list, values, n);

          j = 0;
          for (i = 0; i < r0; i++)
            values[j++] = i;
          for (i = r1 - 1; i >= r0; i--)
            values[j++] = i;
          for (i = r1; i < n; i++)
            values[j++] = i;

          ll_reverse (elemp[r0], elemp[r1]);
          check_list_contents (&list, values, n);

          free_elements (n, elems, elemp, values);
        }
}

/* Tests ll_next_permutation and ll_prev_permutation when the
   permuted values have no duplicates. */
static void
test_permutations_no_dups (void)
{
  const int max_elems = 8;
  int n;

  for (n = 0; n <= max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      int *values;
      int *old_values = xnmalloc (n, sizeof *values);
      int *new_values = xnmalloc (n, sizeof *values);

      size_t n_perms;

      allocate_ascending (n, &list, &elems, NULL, &values);

      n_perms = 1;
      extract_values (&list, old_values, n);
      while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                  compare_elements, &aux_data))
        {
          extract_values (&list, new_values, n);
          check (lexicographical_compare_3way (new_values, n,
                                               old_values, n,
                                               sizeof *new_values,
                                               compare_ints, NULL) > 0);
          memcpy (old_values, new_values, (n) * sizeof *old_values);
          n_perms++;
        }
      check (n_perms == factorial (n));
      check_list_contents (&list, values, n);

      n_perms = 1;
      ll_reverse (ll_head (&list), ll_null (&list));
      extract_values (&list, old_values, n);
      while (ll_prev_permutation (ll_head (&list), ll_null (&list),
                                  compare_elements, &aux_data))
        {
          extract_values (&list, new_values, n);
          check (lexicographical_compare_3way (new_values, n,
                                               old_values, n,
                                               sizeof *new_values,
                                               compare_ints, NULL) < 0);
          memcpy (old_values, new_values, (n) * sizeof *old_values);
          n_perms++;
        }
      check (n_perms == factorial (n));
      ll_reverse (ll_head (&list), ll_null (&list));
      check_list_contents (&list, values, n);

      free_elements (n, elems, NULL, values);
      free (old_values);
      free (new_values);
    }
}

/* Tests ll_next_permutation and ll_prev_permutation when the
   permuted values contain duplicates. */
static void
test_permutations_with_dups (void)
{
  const int max_elems = 8;
  const int max_dup = 3;
  const int repetitions = 1024;

  for (int repeat = 0; repeat < repetitions; repeat++)
    for (int n_elems = 0; n_elems < max_elems; n_elems++)
      {
        struct ll_list list;
        struct element **elems;
        int *values;
        int *old_values = xnmalloc (max_elems, sizeof *values);
        int *new_values = xnmalloc (max_elems, sizeof *values);

        unsigned int n_permutations;
        int left = n_elems;
        int value = 0;

        allocate_elements (n_elems, &list, &elems, NULL, &values);

        value = 0;
        while (left > 0)
          {
            int max = left < max_dup ? left : max_dup;
            int n = rand () % max + 1;
            while (n-- > 0)
              {
                int idx = n_elems - left--;
                values[idx] = elems[idx]->x = value;
              }
            value++;
          }

        n_permutations = 1;
        extract_values (&list, old_values, n_elems);
        while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                    compare_elements, &aux_data))
          {
            extract_values (&list, new_values, n_elems);
            check (lexicographical_compare_3way (new_values, n_elems,
                                                 old_values, n_elems,
                                                 sizeof *new_values,
                                                 compare_ints, NULL) > 0);
            memcpy (old_values, new_values, n_elems * sizeof *old_values);
            n_permutations++;
          }
        check (n_permutations == expected_perms (values, n_elems));
        check_list_contents (&list, values, n_elems);

        n_permutations = 1;
        ll_reverse (ll_head (&list), ll_null (&list));
        extract_values (&list, old_values, n_elems);
        while (ll_prev_permutation (ll_head (&list), ll_null (&list),
                                    compare_elements, &aux_data))
          {
            extract_values (&list, new_values, n_elems);
            check (lexicographical_compare_3way (new_values, n_elems,
                                                 old_values, n_elems,
                                                 sizeof *new_values,
                                                 compare_ints, NULL) < 0);
            n_permutations++;
          }
        ll_reverse (ll_head (&list), ll_null (&list));
        check (n_permutations == expected_perms (values, n_elems));
        check_list_contents (&list, values, n_elems);

        free_elements (n_elems, elems, NULL, values);
        free (old_values);
        free (new_values);
      }
}

/* Tests ll_merge when no equal values are to be merged. */
static void
test_merge_no_dups (void)
{
  const int max_elems = 8;
  const int max_filler = 3;

  int n_merges, pattern, pfx, gap, sfx, order;

  for (n_merges = 0; n_merges < max_elems; n_merges++)
    for (pattern = 0; pattern <= (1 << n_merges); pattern++)
      for (pfx = 0; pfx < max_filler; pfx++)
        for (gap = 0; gap < max_filler; gap++)
          for (sfx = 0; sfx < max_filler; sfx++)
            for (order = 0; order < 2; order++)
              {
                struct ll_list list;
                struct element **elems;
                struct ll **elemp;
                int *values;

                int n_lists = pfx + n_merges + gap + sfx;
                int a0, a1, b0, b1;
                int i, j;

                allocate_elements (n_lists, &list,
                                   &elems, &elemp, &values);

                j = 0;
                for (i = 0; i < pfx; i++)
                  elems[j++]->x = 100 + i;
                a0 = j;
                for (i = 0; i < n_merges; i++)
                  if (pattern & (1u << i))
                    elems[j++]->x = i;
                a1 = j;
                for (i = 0; i < gap; i++)
                  elems[j++]->x = 200 + i;
                b0 = j;
                for (i = 0; i < n_merges; i++)
                  if (!(pattern & (1u << i)))
                    elems[j++]->x = i;
                b1 = j;
                for (i = 0; i < sfx; i++)
                  elems[j++]->x = 300 + i;
                check (n_lists == j);

                j = 0;
                for (i = 0; i < pfx; i++)
                  values[j++] = 100 + i;
                if (order == 0)
                  for (i = 0; i < n_merges; i++)
                    values[j++] = i;
                for (i = 0; i < gap; i++)
                  values[j++] = 200 + i;
                if (order == 1)
                  for (i = 0; i < n_merges; i++)
                    values[j++] = i;
                for (i = 0; i < sfx; i++)
                  values[j++] = 300 + i;
                check (n_lists == j);

                if (order == 0)
                  ll_merge (elemp[a0], elemp[a1], elemp[b0], elemp[b1],
                            compare_elements, &aux_data);
                else
                  ll_merge (elemp[b0], elemp[b1], elemp[a0], elemp[a1],
                            compare_elements, &aux_data);

                check_list_contents (&list, values, n_lists);

                free_elements (n_lists, elems, elemp, values);
              }
}

/* Tests ll_merge when equal values are to be merged. */
static void
test_merge_with_dups (void)
{
  const int max_elems = 8;

  int n, merge_pat, inc_pat, order;

  for (n = 0; n <= max_elems; n++)
    for (merge_pat = 0; merge_pat <= (1 << n); merge_pat++)
      for (inc_pat = 0; inc_pat <= (1 << n); inc_pat++)
        for (order = 0; order < 2; order++)
          {
            struct ll_list list;
            struct element **elems;
            struct ll **elemp;
            int *values;

            int mid;
            int i, j, k;

            allocate_elements (n, &list, &elems, &elemp, &values);

            j = 0;
            for (i = k = 0; i < n; i++)
              {
                if (merge_pat & (1u << i))
                  elems[j++]->x = k;
                if (inc_pat & (1u << i))
                  k++;
              }
            mid = j;
            for (i = k = 0; i < n; i++)
              {
                if (!(merge_pat & (1u << i)))
                  elems[j++]->x = k;
                if (inc_pat & (1u << i))
                  k++;
              }
            check (n == j);

            if (order == 0)
              {
                for (i = 0; i < n; i++)
                  elems[i]->y = i;
              }
            else
              {
                for (i = 0; i < mid; i++)
                  elems[i]->y = 100 + i;
                for (i = mid; i < n; i++)
                  elems[i]->y = i;
              }

            j = 0;
            for (i = k = 0; i < n; i++)
              {
                values[j++] = k;
                if (inc_pat & (1u << i))
                  k++;
              }
            check (n == j);

            if (order == 0)
              ll_merge (elemp[0], elemp[mid], elemp[mid], elemp[n],
                        compare_elements, &aux_data);
            else
              ll_merge (elemp[mid], elemp[n], elemp[0], elemp[mid],
                        compare_elements, &aux_data);

            check_list_contents (&list, values, n);
            check (ll_is_sorted (ll_head (&list), ll_null (&list),
                                 compare_elements_x_y, &aux_data));

            free_elements (n, elems, elemp, values);
          }
}

/* Tests ll_sort on all permutations up to a maximum number of
   elements. */
static void
test_sort_exhaustive (void)
{
  const int max_elems = 8;
  int n;

  for (n = 0; n <= max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      int *values;

      struct element **perm_elems;
      int *perm_values;

      size_t n_perms;

      allocate_ascending (n, &list, &elems, NULL, &values);
      allocate_elements (n, NULL, &perm_elems, NULL, &perm_values);

      n_perms = 1;
      while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                  compare_elements, &aux_data))
        {
          struct ll_list perm_list;
          int j;

          extract_values (&list, perm_values, n);
          ll_init (&perm_list);
          for (j = 0; j < n; j++)
            {
              perm_elems[j]->x = perm_values[j];
              ll_push_tail (&perm_list, &perm_elems[j]->ll);
            }
          ll_sort (ll_head (&perm_list), ll_null (&perm_list),
                   compare_elements, &aux_data);
          check_list_contents (&perm_list, values, n);
          check (ll_is_sorted (ll_head (&perm_list), ll_null (&perm_list),
                               compare_elements, &aux_data));
          n_perms++;
        }
      check (n_perms == factorial (n));

      free_elements (n, elems, NULL, values);
      free_elements (n, perm_elems, NULL, perm_values);
    }
}

/* Tests that ll_sort is stable in the presence of equal
   values. */
static void
test_sort_stable (void)
{
  const int max_elems = 6;
  int n, inc_pat;

  for (n = 0; n <= max_elems; n++)
    for (inc_pat = 0; inc_pat <= 1 << n; inc_pat++)
      {
        struct ll_list list;
        struct element **elems;
        int *values;

        struct element **perm_elems;
        int *perm_values;

        size_t n_perms;
        int i, j;

        allocate_elements (n, &list, &elems, NULL, &values);
        allocate_elements (n, NULL, &perm_elems, NULL, &perm_values);

        j = 0;
        for (i = 0; i < n; i++)
          {
            elems[i]->x = values[i] = j;
            if (inc_pat & (1 << i))
              j++;
            elems[i]->y = i;
          }

        n_perms = 1;
        while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                    compare_elements_y, &aux_data))
          {
            struct ll_list perm_list;

            extract_values (&list, perm_values, n);
            ll_init (&perm_list);
            for (i = 0; i < n; i++)
              {
                perm_elems[i]->x = perm_values[i];
                perm_elems[i]->y = i;
                ll_push_tail (&perm_list, &perm_elems[i]->ll);
              }
            ll_sort (ll_head (&perm_list), ll_null (&perm_list),
                     compare_elements, &aux_data);
            check_list_contents (&perm_list, values, n);
            check (ll_is_sorted (ll_head (&perm_list), ll_null (&perm_list),
                                 compare_elements_x_y, &aux_data));
            n_perms++;
          }
        check (n_perms == factorial (n));

        free_elements (n, elems, NULL, values);
        free_elements (n, perm_elems, NULL, perm_values);
      }
}

/* Tests that ll_sort does not disturb elements outside the
   range sorted. */
static void
test_sort_subset (void)
{
  const int max_elems = 8;

  int n, r0, r1, repeat;

  for (n = 0; n <= max_elems; n++)
    for (repeat = 0; repeat < 100; repeat++)
      for (r0 = 0; r0 <= n; r0++)
        for (r1 = r0; r1 <= n; r1++)
          {
            struct ll_list list;
            struct element **elems;
            struct ll **elemp;
            int *values;

            allocate_random (n, &list, &elems, &elemp, &values);

            qsort (&values[r0], r1 - r0, sizeof *values, compare_ints_noaux);
            ll_sort (elemp[r0], elemp[r1], compare_elements, &aux_data);
            check_list_contents (&list, values, n);

            free_elements (n, elems, elemp, values);
          }
}

/* Tests that ll_sort works with large lists. */
static void
test_sort_big (void)
{
  const int max_elems = 1024;

  int n;

  for (n = 0; n < max_elems; n++)
    {
      struct ll_list list;
      struct element **elems;
      int *values;

      allocate_random (n, &list, &elems, NULL, &values);

      qsort (values, n, sizeof *values, compare_ints_noaux);
      ll_sort (ll_head (&list), ll_null (&list), compare_elements, &aux_data);
      check_list_contents (&list, values, n);

      free_elements (n, elems, NULL, values);
    }
}

/* Tests ll_unique. */
static void
test_unique (void)
{
  const int max_elems = 10;

  int *ascending = xnmalloc (max_elems, sizeof *ascending);

  int n, inc_pat, i, j, unique_values;

  for (i = 0; i < max_elems; i++)
    ascending[i] = i;

  for (n = 0; n < max_elems; n++)
    for (inc_pat = 0; inc_pat < (1 << n); inc_pat++)
      {
        struct ll_list list, dups;
        struct element **elems;
        int *values;

        allocate_elements (n, &list, &elems, NULL, &values);

        j = unique_values = 0;
        for (i = 0; i < n; i++)
          {
            unique_values = j + 1;
            elems[i]->x = values[i] = j;
            if (inc_pat & (1 << i))
              j++;
          }
        check_list_contents (&list, values, n);

        ll_init (&dups);
        check (ll_unique (ll_head (&list), ll_null (&list), ll_null (&dups),
                          compare_elements, &aux_data)
               == (size_t) unique_values);
        check_list_contents (&list, ascending, unique_values);

        ll_splice (ll_null (&list), ll_head (&dups), ll_null (&dups));
        ll_sort (ll_head (&list), ll_null (&list), compare_elements, &aux_data);
        check_list_contents (&list, values, n);

        free_elements (n, elems, NULL, values);
      }

  free (ascending);
}

/* Tests ll_sort_unique. */
static void
test_sort_unique (void)
{
  const int max_elems = 7;
  int n, inc_pat;

  for (n = 0; n <= max_elems; n++)
    for (inc_pat = 0; inc_pat <= 1 << n; inc_pat++)
      {
        struct ll_list list;
        struct element **elems;
        int *values;

        struct element **perm_elems;
        int *perm_values;

        int n_uniques;
        int *unique_values;

        size_t n_perms;
        int i, j;

        allocate_elements (n, &list, &elems, NULL, &values);
        allocate_elements (n, NULL, &perm_elems, NULL, &perm_values);

        j = n_uniques = 0;
        for (i = 0; i < n; i++)
          {
            elems[i]->x = values[i] = j;
            n_uniques = j + 1;
            if (inc_pat & (1 << i))
              j++;
          }

        unique_values = xnmalloc (n_uniques, sizeof *unique_values);
        for (i = 0; i < n_uniques; i++)
          unique_values[i] = i;

        n_perms = 1;
        while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                    compare_elements, &aux_data))
          {
            struct ll_list perm_list;

            extract_values (&list, perm_values, n);
            ll_init (&perm_list);
            for (i = 0; i < n; i++)
              {
                perm_elems[i]->x = perm_values[i];
                perm_elems[i]->y = i;
                ll_push_tail (&perm_list, &perm_elems[i]->ll);
              }
            ll_sort_unique (ll_head (&perm_list), ll_null (&perm_list), NULL,
                            compare_elements, &aux_data);
            check_list_contents (&perm_list, unique_values, n_uniques);
            check (ll_is_sorted (ll_head (&perm_list), ll_null (&perm_list),
                                 compare_elements_x_y, &aux_data));
            n_perms++;
          }
        check (n_perms == expected_perms (values, n));

        free_elements (n, elems, NULL, values);
        free_elements (n, perm_elems, NULL, perm_values);
        free (unique_values);
      }
}

/* Tests ll_insert_ordered. */
static void
test_insert_ordered (void)
{
  const int max_elems = 6;
  int n, inc_pat;

  for (n = 0; n <= max_elems; n++)
    for (inc_pat = 0; inc_pat <= 1 << n; inc_pat++)
      {
        struct ll_list list;
        struct element **elems;
        int *values;

        struct element **perm_elems;
        int *perm_values;

        size_t n_perms;
        int i, j;

        allocate_elements (n, &list, &elems, NULL, &values);
        allocate_elements (n, NULL, &perm_elems, NULL, &perm_values);

        j = 0;
        for (i = 0; i < n; i++)
          {
            elems[i]->x = values[i] = j;
            if (inc_pat & (1 << i))
              j++;
            elems[i]->y = i;
          }

        n_perms = 1;
        while (ll_next_permutation (ll_head (&list), ll_null (&list),
                                    compare_elements_y, &aux_data))
          {
            struct ll_list perm_list;

            extract_values (&list, perm_values, n);
            ll_init (&perm_list);
            for (i = 0; i < n; i++)
              {
                perm_elems[i]->x = perm_values[i];
                perm_elems[i]->y = i;
                ll_insert_ordered (ll_head (&perm_list), ll_null (&perm_list),
                                   &perm_elems[i]->ll,
                                   compare_elements, &aux_data);
              }
            check (ll_is_sorted (ll_head (&perm_list), ll_null (&perm_list),
                                 compare_elements_x_y, &aux_data));
            n_perms++;
          }
        check (n_perms == factorial (n));

        free_elements (n, elems, NULL, values);
        free_elements (n, perm_elems, NULL, perm_values);
      }
}

/* Tests ll_partition. */
static void
test_partition (void)
{
  const int max_elems = 10;

  int n;
  unsigned int pbase;
  int r0, r1;

  for (n = 0; n < max_elems; n++)
    for (r0 = 0; r0 <= n; r0++)
      for (r1 = r0; r1 <= n; r1++)
        for (pbase = 0; pbase <= (1u << (r1 - r0)); pbase++)
          {
            struct ll_list list;
            struct element **elems;
            struct ll **elemp;
            int *values;

            unsigned int pattern = pbase << r0;
            int i, j;
            int first_false;
            struct ll *part_ll;

            allocate_ascending (n, &list, &elems, &elemp, &values);

            /* Check that ll_find_partition works okay in every
               case.  We use it after partitioning, too, but that
               only tests cases where it returns non-null. */
            for (i = r0; i < r1; i++)
              if (!(pattern & (1u << i)))
                break;
            j = i;
            for (; i < r1; i++)
              if (pattern & (1u << i))
                break;
            part_ll = ll_find_partition (elemp[r0], elemp[r1],
                                         pattern_pred,
                                         &pattern);
            if (i == r1)
              check (part_ll == elemp[j]);
            else
              check (part_ll == NULL);

            /* Figure out expected results. */
            j = 0;
            first_false = -1;
            for (i = 0; i < r0; i++)
              values[j++] = i;
            for (i = r0; i < r1; i++)
              if (pattern & (1u << i))
                values[j++] = i;
            for (i = r0; i < r1; i++)
              if (!(pattern & (1u << i)))
                {
                  if (first_false == -1)
                    first_false = i;
                  values[j++] = i;
                }
            if (first_false == -1)
              first_false = r1;
            for (i = r1; i < n; i++)
              values[j++] = i;
            check (j == n);

            /* Partition and check for expected results. */
            check (ll_partition (elemp[r0], elemp[r1],
                                 pattern_pred, &pattern)
                   == elemp[first_false]);
            check (ll_find_partition (elemp[r0], elemp[r1],
                                      pattern_pred, &pattern)
                   == elemp[first_false]);
            check_list_contents (&list, values, n);
            check ((int) ll_count (&list) == n);

            free_elements (n, elems, elemp, values);
          }
}

/* Main program. */

struct test
  {
    const char *name;
    const char *description;
    void (*function) (void);
  };

static const struct test tests[] =
  {
    {
      "push-pop",
      "push/pop",
      test_push_pop
    },
    {
      "insert-remove",
      "insert/remove",
      test_insert_remove
    },
    {
      "swap",
      "swap",
      test_swap
    },
    {
      "swap-range",
      "swap_range",
      test_swap_range
    },
    {
      "remove-range",
      "remove_range",
      test_remove_range
    },
    {
      "remove-equal",
      "remove_equal",
      test_remove_equal
    },
    {
      "remove-if",
      "remove_if",
      test_remove_if
    },
    {
      "moved",
      "moved",
      test_moved
    },
    {
      "find-equal",
      "find_equal",
      test_find_equal
    },
    {
      "find-if",
      "find_if",
      test_find_if
    },
    {
      "find-adjacent-equal",
      "find_adjacent_equal",
      test_find_adjacent_equal
    },
    {
      "count-range",
      "count_range",
      test_count_range
    },
    {
      "count-equal",
      "count_equal",
      test_count_equal
    },
    {
      "count-if",
      "count_if",
      test_count_if
    },
    {
      "min-max",
      "min/max",
      test_min_max
    },
    {
      "lexicographical-compare-3way",
      "lexicographical_compare_3way",
      test_lexicographical_compare_3way
    },
    {
      "apply",
      "apply",
      test_apply
    },
    {
      "reverse",
      "reverse",
      test_reverse
    },
    {
      "permutations-no-dups",
      "permutations (no dups)",
      test_permutations_no_dups
    },
    {
      "permutations-with-dups",
      "permutations (with dups)",
      test_permutations_with_dups
    },
    {
      "merge-no-dups",
      "merge (no dups)",
      test_merge_no_dups
    },
    {
      "merge-with-dups",
      "merge (with dups)",
      test_merge_with_dups
    },
    {
      "sort-exhaustive",
      "sort (exhaustive)",
      test_sort_exhaustive
    },
    {
      "sort-stable",
      "sort (stability)",
      test_sort_stable
    },
    {
      "sort-subset",
      "sort (subset)",
      test_sort_subset
    },
    {
      "sort-big",
      "sort (big)",
      test_sort_big
    },
    {
      "unique",
      "unique",
      test_unique
    },
    {
      "sort-unique",
      "sort_unique",
      test_sort_unique
    },
    {
      "insert-ordered",
      "insert_ordered",
      test_insert_ordered
    },
    {
      "partition",
      "partition",
      test_partition
    },
  };

enum { N_TESTS = sizeof tests / sizeof *tests };

int
main (int argc, char *argv[])
{
  int i;

  if (argc != 2)
    {
      fprintf (stderr, "exactly one argument required; use --help for help\n");
      return EXIT_FAILURE;
    }
  else if (!strcmp (argv[1], "--help"))
    {
      printf ("%s: test doubly linked list (ll) library\n"
              "usage: %s TEST-NAME\n"
              "where TEST-NAME is one of the following:\n",
              argv[0], argv[0]);
      for (i = 0; i < N_TESTS; i++)
        printf ("  %s\n    %s\n", tests[i].name, tests[i].description);
      return 0;
    }
  else
    {
      for (i = 0; i < N_TESTS; i++)
        if (!strcmp (argv[1], tests[i].name))
          {
            tests[i].function ();
            return 0;
          }

      fprintf (stderr, "unknown test %s; use --help for help\n", argv[1]);
      return EXIT_FAILURE;
    }
}