From 215cb550b912fda67b121581ac2bdfbe766bb98d Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Fri, 10 Jul 2020 13:07:28 -0500
Subject: [PATCH] Fix CSEMatVecOperator test
---
test/test_misc.py | 20 ++++++++++++++------
1 file changed, 14 insertions(+), 6 deletions(-)
diff --git a/test/test_misc.py b/test/test_misc.py
index d43c736b..75c77974 100644
--- a/test/test_misc.py
+++ b/test/test_misc.py
@@ -280,13 +280,21 @@ def test_toy_p2e2e2p(ctx_factory, case):
def test_cse_matvec():
from sumpy.expansion import CSEMatVecOperator
- assignments = [
- ([(0, 2)], []),
- ([], [(0, 3)]),
- ([(1, 1)], []),
- ([(1, 9)], [(2, 7), (1, 5)]),
+ input_coeffs = [
+ [(0, 2)],
+ [],
+ [(1, 1)],
+ [(1, 9)],
]
- op = CSEMatVecOperator(assignments, shape=(4, 2))
+
+ output_coeffs = [
+ [],
+ [(0, 3)],
+ [],
+ [(2, 7), (1, 5)],
+ ]
+
+ op = CSEMatVecOperator(input_coeffs, output_coeffs, shape=(4, 2))
m = np.array([[2, 0], [6, 0], [0, 1], [30, 16]])
vec = np.random.random(2)
--
GitLab