lean2/src/tests/sexpr/sexpr.cpp

148 lines
5 KiB
C++
Raw Normal View History

/*
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 "sexpr.h"
#include "sexpr_funcs.h"
#include "mpq.h"
#include "name.h"
#include "test.h"
using namespace lean;
static void tst1() {
sexpr s1(30, nil());
sexpr s2("name", s1);
std::cout << s2 << "\n";
std::cout << sexpr(s2, s2) << "\n";
lean_assert(len(s2) == 2);
lean_assert(is_nil(nil()));
lean_assert(!is_nil(s2));
lean_assert(is_cons(s2));
lean_assert(is_cons(sexpr(s2, s2)));
lean_assert(is_list(s2));
lean_assert(is_cons(sexpr(s2, s2)));
lean_assert(is_list(sexpr(s2, s2)));
lean_assert(!is_list(sexpr(s2, sexpr(10))));
lean_assert(s2 == sexpr("name", sexpr(30, nil())));
lean_assert(s2 != sexpr("name", sexpr(11.2, nil())));
lean_assert(s2 != sexpr("name", sexpr(20, nil())));
lean_assert(s2 == sexpr("name", sexpr(30, nil())));
lean_assert(cdr(s2) == sexpr(30, nil()));
lean_assert(car(s2) == sexpr("name"));
std::cout << sexpr(name(name("foo"), 1), s2) << "\n";
lean_assert(to_mpq(sexpr(mpq("1/3"))) == mpq(1, 3));
lean_assert(to_int(sexpr(1)) == 1);
lean_assert(is_int(sexpr(1)));
lean_assert(!is_nil(sexpr(2)));
std::cout << sexpr(sexpr(10), sexpr(20)) << "\n";
std::cout << sexpr{10, 20, 30, 40} << "\n";
std::cout << sexpr{"foo", "bla", "tst"} << "\n";
std::cout << sexpr{10.20, 3.2, 4.3} << "\n";
std::cout << sexpr(10.2) << "\n";
std::cout << sexpr{10.2} << "\n";
lean_assert(!is_list(sexpr(10.2)));
lean_assert(is_list(sexpr{10.2}));
lean_assert(len(sexpr{10.2}) == 1);
// list of pairs
std::cout << sexpr{ sexpr(1,2), sexpr(2,3), sexpr(0,1) } << "\n";
// list of lists
std::cout << sexpr{ sexpr{1,2}, sexpr{2,3}, sexpr{0,1} } << "\n";
lean_assert(reverse(sexpr{1, 2, 3}) == (sexpr{3, 2, 1}));
sexpr l = map(sexpr{1, 2, 3},
[](sexpr e) {
return sexpr(to_int(e) + 10);
});
std::cout << l << std::endl;
lean_assert(l == (sexpr{11, 12, 13}));
{
sexpr l = filter(sexpr{10, -2, 3, -1, 0}, [](sexpr e) { return to_int(e) >= 0; });
std::cout << l << std::endl;
lean_assert(l == (sexpr{10, 3, 0}));
}
lean_assert(member(3, sexpr{10, 2, 3, 1}));
lean_assert(!member(3, nil()));
lean_assert(!member(3, sexpr{10, 2, 1}));
lean_assert(append(sexpr{1,2}, sexpr{3,4}) == (sexpr{1,2,3,4}));
lean_assert(append(l, nil()) == l);
lean_assert(append(nil(), l) == l);
lean_assert(contains(sexpr{10,20,-2,0,10}, [](sexpr e) { return to_int(e) < 0; }));
lean_assert(!contains(sexpr{10,20,-2,0,10}, [](sexpr e) { return to_int(e) < -10; }));
lean_assert(eqp(s1, s1));
sexpr s3 = s1;
lean_assert(eqp(s1, s3));
lean_assert(!eqp(sexpr(1), sexpr(1)));
}
static void tst2() {
sexpr a;
a = 2;
lean_assert(a == sexpr(2));
lean_assert(a == 2);
lean_assert(2 == a);
a = 0.125;
lean_assert(a == sexpr(0.125));
lean_assert(a == 0.125);
lean_assert(0.125 == a);
a = "foo";
lean_assert(a == sexpr("foo"));
lean_assert(a == "foo");
lean_assert("foo" == a);
lean_assert(a != "blah");
lean_assert(a != name("foo"));
lean_assert(std::string("foo") == a);
lean_assert(a == std::string("foo"));
a = name(name("foo"), 1);
lean_assert(a == sexpr(name(name("foo"), 1)));
lean_assert(a == name(name("foo"), 1));
lean_assert(name(name("foo"), 1) == a);
a = mpq(1,3);
lean_assert(a == sexpr(mpq(1,3)));
lean_assert(a == mpq(1,3));
lean_assert(mpq(1, 3) == a);
lean_assert(mpq(2, 3) != a);
a = power(mpz(2),100);
lean_assert(a == sexpr(power(mpz(2), 100)));
lean_assert(a == power(mpz(2), 100));
lean_assert(power(mpz(2), 100) == a);
lean_assert(mpq(power(mpz(2), 100)) != a);
lean_assert(sexpr(1, 2) != sexpr(2, 1));
lean_assert(sexpr(1, 2) != sexpr(1, sexpr(2, nil())));
lean_assert(sexpr(1, 2) == sexpr(1, sexpr(2)));
}
static void tst3() {
int sum = 0;
foreach(sexpr{0, 1, 2, 3, 4},
[&](sexpr const & e) { sum += to_int(e); });
lean_assert(sum == 10);
}
static void tst4() {
lean_assert(sexpr(1) < sexpr(2));
lean_assert(sexpr() < sexpr(2));
lean_assert(sexpr("foo") < sexpr(2));
lean_assert(sexpr(3) > sexpr(2));
lean_assert(sexpr(2) < sexpr(1.0));
lean_assert(sexpr("foo") < sexpr(name("foo")));
lean_assert(!(sexpr(name("foo")) < sexpr(name("foo"))));
auto e = sexpr(1);
lean_assert(e == e);
lean_assert((sexpr{1, 2, 3}) <= (sexpr{1, 2, 3}));
lean_assert((sexpr{1, 2, 3}) >= (sexpr{1, 2, 3}));
lean_assert((sexpr{1, 1, 3}) < (sexpr{1, 2, 3}));
lean_assert((sexpr{2, 1, 3}) > (sexpr{1, 2, 3}));
lean_assert((sexpr{2, 1, 3}) > (sexpr("foo")));
lean_assert(sexpr(1, 2) > sexpr(0, 1));
}
int main() {
continue_on_violation(true);
tst1();
tst2();
tst3();
tst4();
return has_violations() ? 1 : 0;
}