123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521 |
- #include <iostream>
- #include <iterator>
- #include "ra/test.hh"
- #include "mpdebug.hh"
- using std::cout, std::endl, std::flush, ra::TestRecorder;
- using complex = std::complex<double>;
- using ra::mp::int_list, ra::int_c;
- struct test_type { int a; };
- std::string typecheck(auto && a)
- {
- return std::source_location::current().function_name();
- };
- int main()
- {
- TestRecorder tr;
- {
- std::array<int, 9> a;
- cout << ra::start(a) << endl;
- ra::Small<int, 9> b;
- cout << b << endl;
- }
- tr.section("initialization that works with either");
- {
- std::array<int, 9> a = {0};
- tr.test_eq(0, a);
- ra::Small<int, 9> b = {0};
- tr.test_eq(0, b);
- }
- tr.section("predicates");
- {
- tr.test(std::is_standard_layout_v<ra::Small<float, 2>>);
- ra::Small<int, 2, 3> a;
- static_assert(std::input_iterator<decltype(a.begin())>);
- }
- tr.section("BUG ambiguous case I");
- {
- double U[] = { 1, 2, 3 };
- ra::Small<ra::Big<double, 1>, 1> u = { {U} };
- tr.test_eq(U, u[0]);
-
-
-
-
- }
-
-
-
-
-
-
-
-
-
- tr.section("= unbraced");
- {
- ra::Small<int, 2> a = {9, 3};
- tr.test_eq(9, a[0]);
- tr.test_eq(3, a[1]);
- }
- tr.section("basic facts about builtin arrays");
- {
- using A0 = double[3];
- using A1 = A0[2];
- using A2 = double[2][3];
- A1 x1 = {{1, 2, 3}, {4, 5, 6}};
- A2 x2 = {{1, 2, 3}, {4, 5, 6}};
- tr.test_eq(2u, std::rank<A1>::value);
- tr.test_eq(2, ra::start(x1).rank());
- tr.test_eq(2u, std::rank<A2>::value);
- tr.test_eq(2, ra::start(x2).rank());
- tr.test_eq(ra::start(x2), x1);
- }
- tr.section("top level generics on builtin arrays");
- {
- int a[3] = {1, 2, 3};
- tr.test_eq(1, ra::rank(a));
- tr.test_eq(3, ra::size(a));
- tr.test_eq(3, ra::shape(a)[0]);
- }
- tr.section("top level generics on builtin arrays II");
- {
- int a[2][3] = {{1, 2, 3}, {4, 5, 6}};
- tr.test_eq(2, ra::rank(a));
- tr.test_eq(6, ra::size(a));
- tr.test_eq(2, ra::shape(a)[0]);
- tr.test_eq(3, ra::shape(a)[1]);
- }
- tr.section("scalar constructors");
- {
- tr.section("scalar paren unbraced");
- {
- ra::Small<int, 2> a(9);
- tr.test_eq(9, a[0]);
- tr.test_eq(9, a[1]);
- }
- tr.section("scalar op unbraced");
- {
- ra::Small<complex, 2> a = 9.;
- tr.test_eq(9., a[0]);
- tr.test_eq(9., a[1]);
- }
- tr.section("variants I wish didn't work :-/ just checking for regressions");
- {
- {
- ra::Small<int, 2> a({9});
- tr.test_eq(9, a[0]);
- tr.test_eq(9, a[1]);
- }
- {
- ra::Small<int, 2> a {9};
- tr.test_eq(9, a[0]);
- tr.test_eq(9, a[1]);
- }
- {
- ra::Small<complex, 2> a = {9.};
- tr.test_eq(9., a[0]);
- tr.test_eq(9., a[1]);
- }
- }
- }
- tr.section("empty");
- {
- ra::Small<int, 0> x;
- tr.test_eq(0, x.size());
- }
- tr.section("rank 0");
- {
- tr.section("default");
- {
- ra::Small<int> x;
- x = 3;
- tr.test_eq(3, x());
- }
- tr.section("rank 0 paren unbraced");
- {
- ra::Small<int> a(9);
- tr.test_eq(9, a());
- }
- tr.section("rank 0 paren braced");
- {
- ra::Small<int> a({9});
- tr.test_eq(9, a());
- }
- tr.section("rank 0 op unbraced");
- {
- ra::Small<int> a = 9;
- tr.test_eq(9, a());
- }
- tr.section("rank 0 op braced");
- {
- ra::Small<int> a = {9};
- tr.test_eq(9, a());
- }
- tr.section("rank 0 raw braced");
- {
- ra::Small<int> a {9};
- tr.test_eq(9, a());
- }
- }
- tr.section("rank 1");
- {
- tr.section("= unbraced");
- {
- ra::Small<int, 2> a = {9, 3};
- tr.test_eq(9, a[0]);
- tr.test_eq(3, a[1]);
- }
- tr.section("raw unbraced");
- {
- ra::Small<int, 2> a {9, 3};
- tr.test_eq(9, a[0]);
- tr.test_eq(3, a[1]);
- }
- tr.section("paren unbraced");
- {
- ra::Small<int, 2> a({9, 3});
- tr.test_eq(9, a[0]);
- tr.test_eq(3, a[1]);
- }
- tr.section("paren raw");
- {
- ra::Small<int, 2> a(9, 3);
- tr.test_eq(9, a[0]);
- tr.test_eq(3, a[1]);
- }
- tr.section("= scalar init");
- {
- ra::Small<int, 2> a = 9;
- tr.test_eq(9, a[0]);
- tr.test_eq(9, a[1]);
- }
- tr.section("= scalar init with non-ra::scalar type");
- {
- ra::Small<test_type, 2> a = test_type { 9 };
- tr.test_eq(9, a[0].a);
- tr.test_eq(9, a[1].a);
- a = test_type { 3 };
- tr.test_eq(3, a[0].a);
- tr.test_eq(3, a[1].a);
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- }
- tr.section("rank 2");
- {
- tr.section("rank 2 (paren unun)");
- {
- ra::Small<double, 2, 2> a({1, 2}, {3, 4});
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(1, 0));
- tr.test_eq(4., a(1, 1));
- }
- tr.section("rank 2 (paren un)");
- {
- ra::Small<double, 2, 2> a({{1, 2}, {3, 4}});
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(1, 0));
- tr.test_eq(4., a(1, 1));
- }
- tr.section("rank 2 (= unbraced)");
- {
- ra::Small<double, 2, 2> a = {{1., 2.}, {3., 4.}};
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(1, 0));
- tr.test_eq(4., a(1, 1));
- }
- tr.section("rank 2 (raw unbraced) I");
- {
- ra::Small<double, 2, 2> a {{1, 2}, {3, 4}};
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(1, 0));
- tr.test_eq(4., a(1, 1));
- }
- tr.section("rank 2 (raw unbraced) II");
- {
- ra::Small<double, 2, 3> a {{1, 2, 3}, {4, 5, 6}};
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(0, 2));
- tr.test_eq(4., a(1, 0));
- tr.test_eq(5., a(1, 1));
- tr.test_eq(6., a(1, 2));
- }
- tr.section("rank 2 (paren raw)");
- {
- ra::Small<double, 2, 2> a({1, 2}, {3, 4});
- tr.test_eq(1., a(0, 0));
- tr.test_eq(2., a(0, 1));
- tr.test_eq(3., a(1, 0));
- tr.test_eq(4., a(1, 1));
- }
- tr.section("size 1, higher rank");
- {
- ra::Small<int, 1, 1> a = {{4}};
- tr.test_eq(4, a(0, 0));
- ra::Small<int, 1, 1> b = {4};
- tr.test_eq(4, b(0, 0));
- }
- tr.section("singular sizes I");
- {
- ra::Small<int, 1, 2> a = {{4, 3}};
- tr.test_eq(4, a(0, 0));
- tr.test_eq(3, a(0, 1));
- ra::Small<int, 1, 2> b = {4, 3};
- tr.test_eq(4, b(0, 0));
- tr.test_eq(3, b(0, 1));
- ra::Small<int, 1, 2> c = {4};
- tr.test_eq(4, c(0, 0));
- tr.test_eq(4, c(0, 1));
-
-
-
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- }
- tr.section("higher rank, unbraced");
- {
- ra::Small<int, 2, 3, 4, 2> a = {{{{ 1, 2}, { 3, 4}, { 5, 6}, { 7, 8}},
- {{ 9, 10}, {11, 12}, {13, 14}, {15, 16}},
- {{17, 18}, {19, 20}, {21, 22}, {23, 24}}},
- {{{25, 26}, {27, 28}, {29, 30}, {31, 32}},
- {{33, 34}, {35, 36}, {37, 38}, {39, 40}},
- {{41, 42}, {43, 44}, {45, 46}, {47, 48}}}};
-
- ra::Small<int, 2, 3, 4, 2> b = 0;
- ra::SmallView<int, int_list<48>, int_list<1>> c(b.data());
- c = ra::iota(48, 1);
- tr.test_eq(b, a);
- tr.test_eq(2, ra::shape(a, 0));
- tr.test_eq(3, ra::shape(a, 1));
- tr.test_eq(4, ra::shape(a, 2));
- tr.test_eq(2, ra::shape(a, 3));
- }
- tr.section("item constructor");
- {
- ra::Small<int, 2, 2> a = {{1, 2}, ra::iota(2, 33)};
- tr.test_eq(1, a(0, 0));
- tr.test_eq(2, a(0, 1));
- tr.test_eq(33, a(1, 0));
- tr.test_eq(34, a(1, 1));
- }
- tr.section("raveling constructor");
- {
- ra::Small<int, 2, 2> a = {1, 2, 3, 4};
- tr.test_eq(1, a(0, 0));
- tr.test_eq(2, a(0, 1));
- tr.test_eq(3, a(1, 0));
- tr.test_eq(4, a(1, 1));
- }
- tr.section("raveling constructor from iterators");
- {
- int AA[4] = { 1, 2, 3, 4 };
- auto a = ra::ravel_from_iterators<ra::Small<int, 2, 2>>(AA, AA+4);
- tr.test_eq(1, a(0, 0));
- tr.test_eq(2, a(0, 1));
- tr.test_eq(3, a(1, 0));
- tr.test_eq(4, a(1, 1));
- }
- tr.section("nested Small I");
- {
- ra::Small<ra::Small<int, 2>, 2> a = {{1, 2}, {3, 4}};
- tr.test_eq(1, a(0)(0));
- tr.test_eq(2, a(0)(1));
- tr.test_eq(3, a(1)(0));
- tr.test_eq(4, a(1)(1));
- }
- tr.section("nested Small II");
- {
- auto test = [&](auto && a)
- {
- tr.test_eq(1, a(0)(0, 0));
- tr.test_eq(2, a(0)(0, 1));
- tr.test_eq(3, a(0)(0, 2));
- tr.test_eq(4, a(0)(1, 0));
- tr.test_eq(5, a(0)(1, 1));
- tr.test_eq(6, a(0)(1, 2));
- tr.test_eq(10, a(1)(0, 0));
- tr.test_eq(20, a(1)(0, 1));
- tr.test_eq(30, a(1)(0, 2));
- tr.test_eq(40, a(1)(1, 0));
- tr.test_eq(50, a(1)(1, 1));
- tr.test_eq(60, a(1)(1, 2));
- };
- using In = ra::Small<int, 2, 3>;
- ra::Small<In, 2> a = {{{1, 2, 3}, {4, 5, 6}}, {{10, 20, 30}, {40, 50, 60}}};
- test(a);
- ra::Small<In, 2> b = {In {{1, 2, 3}, {4, 5, 6}}, In {{10, 20, 30}, {40, 50, 60}}};
- test(b);
- int x[2][3] = {{1, 2, 3}, {4, 5, 6}};
- int y[2][3] = {{10, 20, 30}, {40, 50, 60}};
- ra::Small<In, 2> c = {x, y};
- test(c);
- }
- tr.section("operator=");
- {
- tr.section("operator= rank 1");
- {
- ra::Small<complex, 2> a { 3, 4 };
- a = complex(99.);
- tr.test_eq(99., a[0]);
- tr.test_eq(99., a[1]);
- a = 88.;
- tr.test_eq(88., a[0]);
- tr.test_eq(88., a[1]);
- a += 1.;
- tr.test_eq(89., a[0]);
- tr.test_eq(89., a[1]);
- }
- tr.section("operator= rank 2 unbraced");
- {
- ra::Small<int, 2, 2> a = 0;
- a = {{1, 2}, {3, 4}};
- tr.test_eq(1, a(0, 0));
- tr.test_eq(2, a(0, 1));
- tr.test_eq(3, a(1, 0));
- tr.test_eq(4, a(1, 1));
- }
-
-
-
-
-
-
-
-
-
- tr.section("operator= of view into view");
- {
- {
- ra::Small<int, 2, 2> a = {{1, 2}, {3, 4}};
- ra::SmallView<int, int_list<2>, int_list<1>> a0 = a(0);
- ra::SmallView<int, int_list<2>, int_list<1>> a1 = a(1);
- a0 = a1;
- tr.test_eq(ra::Small<int, 2, 2> {{3, 4}, {3, 4}}, a);
- }
- {
- ra::Small<int, 2, 2> a = {{1, 2}, {3, 4}};
- ra::SmallView<int, int_list<2>, int_list<1>> a0 = a(0);
- a0 = a(1);
- tr.test_eq(ra::Small<int, 2, 2> {{3, 4}, {3, 4}}, a);
- }
- }
- tr.section("repeat a test from test/big-0 [ra38]");
- {
- ra::Small<double, 2> A = {1, 2};
- ra::Small<double, 2> X = {0, 0};
- X(ra::all) = A();
- tr.test_eq(ra::start({1, 2}), X);
- }
- }
-
-
-
-
-
-
-
- {
- ra::Small<int, 3> a = 0;
- ra::Small<int, 3> b = {9, 8, 7};
- ra::Small<int, 3> c = {9, 8, 7};
- auto pa = a.data();
- tr.test_eq(a, 0);
- cout << typecheck(a()) << endl;
- a() = b();
- tr.test_eq(a, c);
- tr.test_eq(ra::scalar(a.data()), ra::scalar(pa));
- a = 0;
- tr.test_eq(a, 0);
- cout << typecheck(a()) << endl;
- a = b;
- tr.test_eq(a, c);
- tr.test_eq(ra::scalar(a.data()), ra::scalar(pa));
- }
- {
- ra::Small<int, 3, 4> a = 0;
- tr.test_eq(2, int_c<rank(a)>::value);
- tr.test_eq(3, int_c<shape(a)[0]>::value);
- tr.test_eq(4, int_c<shape(a)[1]>::value);
- tr.test_eq(12, int_c<ra::size(a)>::value);
- }
- return tr.summary();
- }
|