Better ASSERT macro.
[gnulib.git] / tests / test-rbtree_list.c
1 /* Test of sequential list data type implementation.
2    Copyright (C) 2006-2007 Free Software Foundation, Inc.
3    Written by Bruno Haible <bruno@clisp.org>, 2006.
4
5    This program is free software; you can redistribute it and/or modify
6    it under the terms of the GNU General Public License as published by
7    the Free Software Foundation; either version 2, or (at your option)
8    any later version.
9
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU General Public License for more details.
14
15    You should have received a copy of the GNU General Public License
16    along with this program; if not, write to the Free Software Foundation,
17    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
18
19 #ifdef HAVE_CONFIG_H
20 # include <config.h>
21 #endif
22
23 #include "gl_rbtree_list.h"
24
25 #include <stdio.h>
26 #include <stdlib.h>
27
28 #include "gl_array_list.h"
29 #include "progname.h"
30
31 extern void gl_rbtree_list_check_invariants (gl_list_t list);
32
33 static const char *objects[15] =
34   {
35     "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o"
36   };
37
38 #define SIZEOF(array) (sizeof (array) / sizeof (array[0]))
39 #define ASSERT(expr) \
40   do                                                                         \
41     {                                                                        \
42       if (!(expr))                                                           \
43         {                                                                    \
44           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
45           abort ();                                                          \
46         }                                                                    \
47     }                                                                        \
48   while (0)
49 #define RANDOM(n) (rand () % (n))
50 #define RANDOM_OBJECT() objects[RANDOM (SIZEOF (objects))]
51
52 static void
53 check_equals (gl_list_t list1, gl_list_t list2)
54 {
55   size_t n, i;
56
57   n = gl_list_size (list1);
58   ASSERT (n == gl_list_size (list2));
59   for (i = 0; i < n; i++)
60     {
61       ASSERT (gl_list_get_at (list1, i) == gl_list_get_at (list2, i));
62     }
63 }
64
65 static void
66 check_all (gl_list_t list1, gl_list_t list2, gl_list_t list3)
67 {
68   gl_rbtree_list_check_invariants (list2);
69   gl_rbtree_list_check_invariants (list3);
70   check_equals (list1, list2);
71   check_equals (list1, list3);
72 }
73
74 int
75 main (int argc, char *argv[])
76 {
77   gl_list_t list1, list2, list3;
78
79   set_program_name (argv[0]);
80
81   /* Allow the user to provide a non-default random seed on the command line.  */
82   if (argc > 1)
83     srand (atoi (argv[1]));
84
85   {
86     size_t initial_size = RANDOM (50);
87     const void **contents =
88       (const void **) malloc (initial_size * sizeof (const void *));
89     size_t i;
90     unsigned int repeat;
91
92     for (i = 0; i < initial_size; i++)
93       contents[i] = RANDOM_OBJECT ();
94
95     /* Create list1.  */
96     list1 = gl_list_create (GL_ARRAY_LIST, NULL, NULL, NULL, true,
97                             initial_size, contents);
98     /* Create list2.  */
99     list2 = gl_list_create_empty (GL_RBTREE_LIST, NULL, NULL, NULL, true);
100     for (i = 0; i < initial_size; i++)
101       gl_list_add_last (list2, contents[i]);
102
103     /* Create list3.  */
104     list3 = gl_list_create (GL_RBTREE_LIST, NULL, NULL, NULL, true,
105                             initial_size, contents);
106
107     check_all (list1, list2, list3);
108
109     for (repeat = 0; repeat < 10000; repeat++)
110       {
111         unsigned int operation = RANDOM (16);
112         switch (operation)
113           {
114           case 0:
115             if (gl_list_size (list1) > 0)
116               {
117                 size_t index = RANDOM (gl_list_size (list1));
118                 const char *obj = RANDOM_OBJECT ();
119                 gl_list_node_t node1, node2, node3;
120
121                 node1 = gl_list_set_at (list1, index, obj);
122                 ASSERT (gl_list_get_at (list1, index) == obj);
123                 ASSERT (gl_list_node_value (list1, node1) == obj);
124
125                 node2 = gl_list_set_at (list2, index, obj);
126                 ASSERT (gl_list_get_at (list2, index) == obj);
127                 ASSERT (gl_list_node_value (list2, node2) == obj);
128
129                 node3 = gl_list_set_at (list3, index, obj);
130                 ASSERT (gl_list_get_at (list3, index) == obj);
131                 ASSERT (gl_list_node_value (list3, node3) == obj);
132
133                 if (index > 0)
134                   {
135                     ASSERT (gl_list_node_value (list1, gl_list_previous_node (list1, node1))
136                             == gl_list_get_at (list1, index - 1));
137                     ASSERT (gl_list_node_value (list2, gl_list_previous_node (list3, node3))
138                             == gl_list_get_at (list2, index - 1));
139                     ASSERT (gl_list_node_value (list3, gl_list_previous_node (list3, node3))
140                             == gl_list_get_at (list2, index - 1));
141                   }
142                 if (index + 1 < gl_list_size (list1))
143                   {
144                     ASSERT (gl_list_node_value (list1, gl_list_next_node (list1, node1))
145                             == gl_list_get_at (list1, index + 1));
146                     ASSERT (gl_list_node_value (list2, gl_list_next_node (list3, node3))
147                             == gl_list_get_at (list2, index + 1));
148                     ASSERT (gl_list_node_value (list3, gl_list_next_node (list3, node3))
149                             == gl_list_get_at (list2, index + 1));
150                   }
151               }
152             break;
153           case 1:
154             {
155               const char *obj = RANDOM_OBJECT ();
156               gl_list_node_t node1, node2, node3;
157               node1 = gl_list_search (list1, obj);
158               node2 = gl_list_search (list2, obj);
159               node3 = gl_list_search (list3, obj);
160               if (node1 == NULL)
161                 {
162                   ASSERT (node2 == NULL);
163                   ASSERT (node3 == NULL);
164                 }
165               else
166                 {
167                   ASSERT (node2 != NULL);
168                   ASSERT (node3 != NULL);
169                   ASSERT (gl_list_node_value (list1, node1) == obj);
170                   ASSERT (gl_list_node_value (list2, node2) == obj);
171                   ASSERT (gl_list_node_value (list3, node3) == obj);
172                 }
173             }
174             break;
175           case 2:
176             {
177               const char *obj = RANDOM_OBJECT ();
178               size_t index1, index2, index3;
179               index1 = gl_list_indexof (list1, obj);
180               index2 = gl_list_indexof (list2, obj);
181               index3 = gl_list_indexof (list3, obj);
182               if (index1 == (size_t)(-1))
183                 {
184                   ASSERT (index2 == (size_t)(-1));
185                   ASSERT (index3 == (size_t)(-1));
186                 }
187               else
188                 {
189                   ASSERT (index2 != (size_t)(-1));
190                   ASSERT (index3 != (size_t)(-1));
191                   ASSERT (gl_list_get_at (list1, index1) == obj);
192                   ASSERT (gl_list_get_at (list2, index2) == obj);
193                   ASSERT (gl_list_get_at (list3, index3) == obj);
194                   ASSERT (index2 == index1);
195                   ASSERT (index3 == index1);
196                 }
197             }
198             break;
199           case 3: /* add 1 element */
200             {
201               const char *obj = RANDOM_OBJECT ();
202               gl_list_node_t node1, node2, node3;
203               node1 = gl_list_add_first (list1, obj);
204               node2 = gl_list_add_first (list2, obj);
205               node3 = gl_list_add_first (list3, obj);
206               ASSERT (gl_list_node_value (list1, node1) == obj);
207               ASSERT (gl_list_node_value (list2, node2) == obj);
208               ASSERT (gl_list_node_value (list3, node3) == obj);
209               ASSERT (gl_list_get_at (list1, 0) == obj);
210               ASSERT (gl_list_get_at (list2, 0) == obj);
211               ASSERT (gl_list_get_at (list3, 0) == obj);
212             }
213             break;
214           case 4: /* add 1 element */
215             {
216               const char *obj = RANDOM_OBJECT ();
217               gl_list_node_t node1, node2, node3;
218               node1 = gl_list_add_last (list1, obj);
219               node2 = gl_list_add_last (list2, obj);
220               node3 = gl_list_add_last (list3, obj);
221               ASSERT (gl_list_node_value (list1, node1) == obj);
222               ASSERT (gl_list_node_value (list2, node2) == obj);
223               ASSERT (gl_list_node_value (list3, node3) == obj);
224               ASSERT (gl_list_get_at (list1, gl_list_size (list1) - 1) == obj);
225               ASSERT (gl_list_get_at (list2, gl_list_size (list2) - 1) == obj);
226               ASSERT (gl_list_get_at (list3, gl_list_size (list3) - 1) == obj);
227             }
228             break;
229           case 5: /* add 3 elements */
230             {
231               const char *obj0 = RANDOM_OBJECT ();
232               const char *obj1 = RANDOM_OBJECT ();
233               const char *obj2 = RANDOM_OBJECT ();
234               gl_list_node_t node1, node2, node3;
235               node1 = gl_list_add_first (list1, obj2);
236               node1 = gl_list_add_before (list1, node1, obj0);
237               node1 = gl_list_add_after (list1, node1, obj1);
238               node2 = gl_list_add_first (list2, obj2);
239               node2 = gl_list_add_before (list2, node2, obj0);
240               node2 = gl_list_add_after (list2, node2, obj1);
241               node3 = gl_list_add_first (list3, obj2);
242               node3 = gl_list_add_before (list3, node3, obj0);
243               node3 = gl_list_add_after (list3, node3, obj1);
244               ASSERT (gl_list_node_value (list1, node1) == obj1);
245               ASSERT (gl_list_node_value (list2, node2) == obj1);
246               ASSERT (gl_list_node_value (list3, node3) == obj1);
247               ASSERT (gl_list_get_at (list1, 0) == obj0);
248               ASSERT (gl_list_get_at (list1, 1) == obj1);
249               ASSERT (gl_list_get_at (list1, 2) == obj2);
250               ASSERT (gl_list_get_at (list2, 0) == obj0);
251               ASSERT (gl_list_get_at (list2, 1) == obj1);
252               ASSERT (gl_list_get_at (list2, 2) == obj2);
253               ASSERT (gl_list_get_at (list3, 0) == obj0);
254               ASSERT (gl_list_get_at (list3, 1) == obj1);
255               ASSERT (gl_list_get_at (list3, 2) == obj2);
256             }
257             break;
258           case 6: /* add 1 element */
259             {
260               size_t index = RANDOM (gl_list_size (list1) + 1);
261               const char *obj = RANDOM_OBJECT ();
262               gl_list_node_t node1, node2, node3;
263               node1 = gl_list_add_at (list1, index, obj);
264               node2 = gl_list_add_at (list2, index, obj);
265               node3 = gl_list_add_at (list3, index, obj);
266               ASSERT (gl_list_get_at (list1, index) == obj);
267               ASSERT (gl_list_node_value (list1, node1) == obj);
268               ASSERT (gl_list_get_at (list2, index) == obj);
269               ASSERT (gl_list_node_value (list2, node2) == obj);
270               ASSERT (gl_list_get_at (list3, index) == obj);
271               ASSERT (gl_list_node_value (list3, node3) == obj);
272               if (index > 0)
273                 {
274                   ASSERT (gl_list_node_value (list1, gl_list_previous_node (list1, node1))
275                           == gl_list_get_at (list1, index - 1));
276                   ASSERT (gl_list_node_value (list2, gl_list_previous_node (list3, node3))
277                           == gl_list_get_at (list2, index - 1));
278                   ASSERT (gl_list_node_value (list3, gl_list_previous_node (list3, node3))
279                           == gl_list_get_at (list2, index - 1));
280                 }
281               if (index + 1 < gl_list_size (list1))
282                 {
283                   ASSERT (gl_list_node_value (list1, gl_list_next_node (list1, node1))
284                           == gl_list_get_at (list1, index + 1));
285                   ASSERT (gl_list_node_value (list2, gl_list_next_node (list3, node3))
286                           == gl_list_get_at (list2, index + 1));
287                   ASSERT (gl_list_node_value (list3, gl_list_next_node (list3, node3))
288                           == gl_list_get_at (list2, index + 1));
289                 }
290             }
291             break;
292           case 7: case 8: /* remove 1 element */
293             if (gl_list_size (list1) > 0)
294               {
295                 size_t n = gl_list_size (list1);
296                 const char *obj = gl_list_get_at (list1, RANDOM (n));
297                 gl_list_node_t node1, node2, node3;
298                 node1 = gl_list_search (list1, obj);
299                 node2 = gl_list_search (list2, obj);
300                 node3 = gl_list_search (list3, obj);
301                 ASSERT (node1 != NULL);
302                 ASSERT (node2 != NULL);
303                 ASSERT (node3 != NULL);
304                 ASSERT (gl_list_remove_node (list1, node1));
305                 ASSERT (gl_list_remove_node (list2, node2));
306                 ASSERT (gl_list_remove_node (list3, node3));
307                 ASSERT (gl_list_size (list1) == n - 1);
308               }
309             break;
310           case 9: case 10: /* remove 1 element */
311             if (gl_list_size (list1) > 0)
312               {
313                 size_t n = gl_list_size (list1);
314                 size_t index = RANDOM (n);
315                 ASSERT (gl_list_remove_at (list1, index));
316                 ASSERT (gl_list_remove_at (list2, index));
317                 ASSERT (gl_list_remove_at (list3, index));
318                 ASSERT (gl_list_size (list1) == n - 1);
319               }
320             break;
321           case 11: case 12: /* remove 1 element */
322             if (gl_list_size (list1) > 0)
323               {
324                 size_t n = gl_list_size (list1);
325                 const char *obj = gl_list_get_at (list1, RANDOM (n));
326                 ASSERT (gl_list_remove (list1, obj));
327                 ASSERT (gl_list_remove (list2, obj));
328                 ASSERT (gl_list_remove (list3, obj));
329                 ASSERT (gl_list_size (list1) == n - 1);
330               }
331             break;
332           case 13:
333             if (gl_list_size (list1) > 0)
334               {
335                 size_t n = gl_list_size (list1);
336                 const char *obj = "xyzzy";
337                 ASSERT (!gl_list_remove (list1, obj));
338                 ASSERT (!gl_list_remove (list2, obj));
339                 ASSERT (!gl_list_remove (list3, obj));
340                 ASSERT (gl_list_size (list1) == n);
341               }
342             break;
343           case 14:
344             {
345               size_t n = gl_list_size (list1);
346               gl_list_iterator_t iter1, iter2, iter3;
347               const void *elt;
348               iter1 = gl_list_iterator (list1);
349               iter2 = gl_list_iterator (list2);
350               iter3 = gl_list_iterator (list3);
351               for (i = 0; i < n; i++)
352                 {
353                   ASSERT (gl_list_iterator_next (&iter1, &elt, NULL));
354                   ASSERT (gl_list_get_at (list1, i) == elt);
355                   ASSERT (gl_list_iterator_next (&iter2, &elt, NULL));
356                   ASSERT (gl_list_get_at (list2, i) == elt);
357                   ASSERT (gl_list_iterator_next (&iter3, &elt, NULL));
358                   ASSERT (gl_list_get_at (list3, i) == elt);
359                 }
360               ASSERT (!gl_list_iterator_next (&iter1, &elt, NULL));
361               ASSERT (!gl_list_iterator_next (&iter2, &elt, NULL));
362               ASSERT (!gl_list_iterator_next (&iter3, &elt, NULL));
363               gl_list_iterator_free (&iter1);
364               gl_list_iterator_free (&iter2);
365               gl_list_iterator_free (&iter3);
366             }
367             break;
368           case 15:
369             {
370               size_t end = RANDOM (gl_list_size (list1) + 1);
371               size_t start = RANDOM (end + 1);
372               gl_list_iterator_t iter1, iter2, iter3;
373               const void *elt;
374               iter1 = gl_list_iterator_from_to (list1, start, end);
375               iter2 = gl_list_iterator_from_to (list2, start, end);
376               iter3 = gl_list_iterator_from_to (list3, start, end);
377               for (i = start; i < end; i++)
378                 {
379                   ASSERT (gl_list_iterator_next (&iter1, &elt, NULL));
380                   ASSERT (gl_list_get_at (list1, i) == elt);
381                   ASSERT (gl_list_iterator_next (&iter2, &elt, NULL));
382                   ASSERT (gl_list_get_at (list2, i) == elt);
383                   ASSERT (gl_list_iterator_next (&iter3, &elt, NULL));
384                   ASSERT (gl_list_get_at (list3, i) == elt);
385                 }
386               ASSERT (!gl_list_iterator_next (&iter1, &elt, NULL));
387               ASSERT (!gl_list_iterator_next (&iter2, &elt, NULL));
388               ASSERT (!gl_list_iterator_next (&iter3, &elt, NULL));
389               gl_list_iterator_free (&iter1);
390               gl_list_iterator_free (&iter2);
391               gl_list_iterator_free (&iter3);
392             }
393             break;
394           }
395         check_all (list1, list2, list3);
396       }
397
398     gl_list_free (list1);
399     gl_list_free (list2);
400     gl_list_free (list3);
401     free (contents);
402   }
403
404   return 0;
405 }