123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990 |
- #include <iostream>
- #include "ra/test.hh"
- #include "mpdebug.hh"
- using std::cout, std::endl, std::flush, ra::TestRecorder;
- using real = double;
- int main()
- {
- TestRecorder tr;
- tr.section("resize first dimension");
- {
- auto test = [&tr](auto const & ref, auto & a, int newsize, int testsize)
- {
- a.resize(newsize);
- tr.test_eq(ref.rank(), a.rank());
- tr.test_eq(newsize, a.len(0));
- for (int i=1; i<a.rank(); ++i) {
- tr.test_eq(ref.len(i), a.len(i));
- }
- tr.test_eq(ref(ra::iota(testsize)), a(ra::iota(testsize)));
- };
- tr.section("rank 2");
- {
- ra::Big<int, 2> a({5, 3}, ra::_0 - ra::_1);
- ra::Big<int, 2> ref = a;
- test(ref, a, 5, 5);
- test(ref, a, 8, 5);
- test(ref, a, 3, 3);
- test(ref, a, 5, 3);
- }
- tr.section("rank 1");
- {
- ra::Big<int, 1> a({2}, 3);
- a.resize(4, 9);
- tr.test_eq(3, a[0]);
- tr.test_eq(3, a[1]);
- tr.test_eq(9, a[2]);
- tr.test_eq(9, a[3]);
- }
- tr.section("rank 3");
- {
- ra::Big<int, 3> a({0, 3, 2}, ra::_0 - ra::_1 + ra::_2);
- ra::Big<int, 3> ref0 = a;
- test(ref0, a, 3, 0);
- a = 77.;
- ra::Big<int, 3> ref1 = a;
- test(ref1, a, 5, 3);
- }
- }
- tr.section("full resize");
- {
- ra::Big<int> a({3, 3, 3, 3}, ra::_0 - ra::_1 + ra::_2 - ra::_3);
- ra::Big<int, 1> b = ra::reshape(a, {50});
- a.resize(ra::Small<int, 2>{5, 10});
- tr.test_eq(5, a.len(0));
- tr.test_eq(10, a.len(1));
- tr.test_eq(b, ra::reshape(a, {50}));
- }
- tr.section("push back");
- {
- real check[] = { 2, 3, 4, 7 };
- auto test = [&tr, &check](auto && z)
- {
- tr.test_eq(0, z.len(0));
- tr.test_eq(1, z.step(0));
- for (int i=0; i<4; ++i) {
- z.push_back(check[i]);
- tr.test_eq(i+1, z.size());
- for (int j=0; j<=i; ++j) {
- tr.test_eq(check[j], z[j]);
- }
- }
- };
- test(ra::Big<real, 1>());
- ra::Big<real> z = ra::Big<real, 1>();
- test(z);
- }
- return tr.summary();
- }
|