blob: 6cdc3940b0b4e47d4d2df1320cda3628f5240c33 [file] [log] [blame]
Wenzel Jakob9e0a0562016-05-05 20:33:54 +02001#!/usr/bin/env python
2from __future__ import print_function
3import sys
4sys.path.append('.')
5
6from example import fixed_r, fixed_c
7from example import fixed_passthrough_r, fixed_passthrough_c
8from example import dense_r, dense_c
9from example import dense_passthrough_r, dense_passthrough_c
10from example import sparse_r, sparse_c
11from example import sparse_passthrough_r, sparse_passthrough_c
Ben North4a220912016-07-05 20:03:02 +010012from example import double_row, double_col
Ben North7b8d9e02016-07-05 21:03:19 +010013from example import double_mat_cm, double_mat_rm
Jason Rhinelander5fd50742016-08-03 16:50:22 -040014from example import cholesky1, cholesky2, cholesky3, cholesky4, cholesky5, cholesky6
Jason Rhinelander7de9f6c2016-07-08 17:44:12 -040015try:
16 import numpy as np
Jason Rhinelandereae180c2016-07-12 14:16:46 -040017 import scipy
Jason Rhinelander7de9f6c2016-07-08 17:44:12 -040018except ImportError:
19 # NumPy missing: skip test
20 exit(99)
Wenzel Jakob9e0a0562016-05-05 20:33:54 +020021
22ref = np.array(
23 [[0, 3, 0, 0, 0, 11],
24 [22, 0, 0, 0, 17, 11],
25 [7, 5, 0, 1, 0, 11],
26 [0, 0, 0, 0, 0, 11],
27 [0, 0, 14, 0, 8, 11]])
28
29
30def check(mat):
Ben North150a0fa2016-07-05 19:59:28 +010031 return 'OK' if np.sum(abs(mat - ref)) == 0 else 'NOT OK'
Wenzel Jakob9e0a0562016-05-05 20:33:54 +020032
Ben North676e2982016-07-05 21:46:09 +010033print("should_give_NOT_OK = %s" % check(ref[:, ::-1]))
34
Wenzel Jakob9e0a0562016-05-05 20:33:54 +020035print("fixed_r = %s" % check(fixed_r()))
36print("fixed_c = %s" % check(fixed_c()))
37print("pt_r(fixed_r) = %s" % check(fixed_passthrough_r(fixed_r())))
38print("pt_c(fixed_c) = %s" % check(fixed_passthrough_c(fixed_c())))
39print("pt_r(fixed_c) = %s" % check(fixed_passthrough_r(fixed_c())))
40print("pt_c(fixed_r) = %s" % check(fixed_passthrough_c(fixed_r())))
41
42print("dense_r = %s" % check(dense_r()))
43print("dense_c = %s" % check(dense_c()))
44print("pt_r(dense_r) = %s" % check(dense_passthrough_r(dense_r())))
45print("pt_c(dense_c) = %s" % check(dense_passthrough_c(dense_c())))
46print("pt_r(dense_c) = %s" % check(dense_passthrough_r(dense_c())))
47print("pt_c(dense_r) = %s" % check(dense_passthrough_c(dense_r())))
48
49print("sparse_r = %s" % check(sparse_r()))
50print("sparse_c = %s" % check(sparse_c()))
51print("pt_r(sparse_r) = %s" % check(sparse_passthrough_r(sparse_r())))
52print("pt_c(sparse_c) = %s" % check(sparse_passthrough_c(sparse_c())))
53print("pt_r(sparse_c) = %s" % check(sparse_passthrough_r(sparse_c())))
54print("pt_c(sparse_r) = %s" % check(sparse_passthrough_c(sparse_r())))
Ben North4a220912016-07-05 20:03:02 +010055
56def check_got_vs_ref(got_x, ref_x):
57 return 'OK' if np.array_equal(got_x, ref_x) else 'NOT OK'
58
59counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3))
60first_row = counting_mat[0, :]
61first_col = counting_mat[:, 0]
62
63print("double_row(first_row) = %s" % check_got_vs_ref(double_row(first_row), 2.0 * first_row))
64print("double_col(first_row) = %s" % check_got_vs_ref(double_col(first_row), 2.0 * first_row))
65print("double_row(first_col) = %s" % check_got_vs_ref(double_row(first_col), 2.0 * first_col))
66print("double_col(first_col) = %s" % check_got_vs_ref(double_col(first_col), 2.0 * first_col))
Ben North7b8d9e02016-07-05 21:03:19 +010067
68counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3))
69slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]]
70
71for slice_idx, ref_mat in enumerate(slices):
72 print("double_mat_cm(%d) = %s" % (slice_idx, check_got_vs_ref(double_mat_cm(ref_mat), 2.0 * ref_mat)))
73 print("double_mat_rm(%d) = %s" % (slice_idx, check_got_vs_ref(double_mat_rm(ref_mat), 2.0 * ref_mat)))
Jason Rhinelander5fd50742016-08-03 16:50:22 -040074
75i = 1
76for chol in [cholesky1, cholesky2, cholesky3, cholesky4, cholesky5, cholesky6]:
77 mymat = chol(np.array([[1,2,4], [2,13,23], [4,23,77]]))
78 print("cholesky" + str(i) + " " + ("OK" if (mymat == np.array([[1,0,0], [2,3,0], [4,5,6]])).all() else "NOT OKAY"))
79 i += 1
80