File: wei-ku-1.cpp

package info (click to toggle)
blitz%2B%2B 1%3A1.0.2%2Bds-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 8,568 kB
  • sloc: cpp: 57,803; python: 1,941; fortran: 1,510; f90: 852; makefile: 833; sh: 321
file content (41 lines) | stat: -rw-r--r-- 853 bytes parent folder | download | duplicates (3)
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
#include "testsuite.h"
#include <blitz/array.h>

using namespace blitz;

int main()
{
  using namespace blitz::tensor;
  Range all = Range::all();

  {
    Array<int,5> A(6,5,4,3,2);
    int ld_begin = 1;
    int ld_end = 3;
    Array<int,5> B(6,5,2,3,2);

    A = 11111 + 10000*i + 1000*j + 100*k + 10*l + m;

    B = A(all, all, Range(ld_begin, ld_end-1), all, all);
    // cout << B << endl;    

    BZTEST(B(0,0,0,0,0) == 11211);
    BZTEST(B(1,1,1,1,1) == 22322);
    BZTEST(B(5,4,1,2,1) == 65332);
  }

  {
    Array<int,5> B(10,1,25,6,6);
    Array<int,5> A(10,1,125,6,6);
    int ld_begin = 1, ld_end = 26;
    A = 11111 + 10000*i + 1000*j + 100*k + 10*l + m;
    B = A(all, all, Range(ld_begin, ld_end-1), all, all);

    BZTEST(B(0,0,0,0,0) == 11211);
    BZTEST(B(1,0,1,1,1) == 21322);
    BZTEST(B(5,0,3,4,5) == 61556);
  }

  return 0;
}