lean2/src/tests/util/list.cpp
Leonardo de Moura 26097475fd Use fullpath in #include directives.
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
2013-09-12 20:04:10 -07:00

85 lines
2.4 KiB
C++

/*
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
*/
#include <vector>
#include "util/list.h"
#include "util/list_fn.h"
#include "util/test.h"
using namespace lean;
static void tst1() {
list<int> l;
lean_assert(is_nil(l));
for (int i = 0; i < 10; i++) {
list<int> old = l;
l = list<int>(i, l);
lean_assert(!is_nil(l));
lean_assert(car(l) == i);
lean_assert(is_eqp(cdr(l), old));
}
std::cout << l << "\n";
}
static void tst2() {
std::vector<int> a{10, 20, 30};
list<int> l = to_list(a.begin(), a.end());
std::cout << l << "\n";
lean_assert(head(l) == 10);
lean_assert(head(tail(l)) == 20);
lean_assert(head(tail(tail(l))) == 30);
lean_assert(length(l) == 3);
}
static void tst3() {
int a[3] = {10, 20, 30};
list<int> l = to_list<int*>(a, a+3);
std::cout << l << "\n";
lean_assert(head(l) == 10);
lean_assert(head(tail(l)) == 20);
lean_assert(head(tail(tail(l))) == 30);
lean_assert(length(l) == 3);
lean_assert(length(list<int>()) == 0);
lean_assert(length(list<int>(10, list<int>())) == 1);
lean_assert(length(tail(list<int>(10, list<int>()))) == 0);
}
static void tst4() {
list<int> l1{1, 2, 3};
list<int> l2{1, 2, 3};
lean_assert(l1 == l2);
lean_assert(l1 != cons(1, l2));
lean_assert(l1 != tail(l1));
lean_assert(list<int>() == list<int>());
lean_assert(l2 != list<int>());
lean_assert(l1 != list<int>({1, 2, 3, 4}));
lean_assert(l1 != tail(list<int>{1, 2, 3, 4}));
}
static void tst5() {
for (unsigned n = 0; n < 16; n++) {
buffer<int> tmp;
for (unsigned i = 0; i < n; i++) { tmp.push_back(i); }
buffer<int> tmp2;
to_buffer(to_list(tmp.begin(), tmp.end()), tmp2);
lean_assert(tmp2 == tmp);
list<int> l = to_list(tmp.begin(), tmp.end());
lean_assert(n == 0 || car(reverse(l)) == static_cast<int>(n - 1));
lean_assert(reverse(reverse(l)) == l);
auto p = split(l);
list<int> l2 = append(p.first, p.second);
lean_assert(l2 == l);
lean_assert(-1 <= static_cast<int>(length(p.first)) - static_cast<int>(length(p.second)) && static_cast<int>(length(p.first)) - static_cast<int>(length(p.second)) <= 1);
}
}
int main() {
tst1();
tst2();
tst3();
tst4();
tst5();
return has_violations() ? 1 : 0;
}