matrices QE, QI and Q provided by new function get_sweeper_mats in imex_1st_order sweper class

Author: Daniel Ruprecht

pySDC/sweeper_classes/imex_1st_order.py

@@ -129,7 +129,6 @@ def update_nodes(self):
 
         return None
 
-
     def compute_end_point(self):
         """
         Compute u at the right point of the interval
@@ -155,3 +154,9 @@ def compute_end_point(self):
               L.uend += L.tau[-1]
 
         return None
+
+    def get_sweeper_mats(self):
+      QE = self.QE[1:,1:]
+      QI = self.QI[1:,1:]
+      Q  = self.coll.Qmat[1:,1:]
+      return QE, QI, Q

tests/test_imexsweeper.py

@@ -33,9 +33,7 @@ def setupLevelStepProblem(self):
     return step, level, problem, nnodes
 
   def setupQMatrices(self, level):
-    QE = level.sweep.QE[1:,1:]
-    QI = level.sweep.QI[1:,1:]
-    Q  = level.sweep.coll.Qmat[1:,1:]
+    QE, QI, Q = level.sweep.get_sweeper_mats()
     return QE, QI, Q
 
   def setupSweeperMatrices(self, step, level, problem):