Test rewrite

.github/workflows/ci-build.yml

@@ -52,7 +52,7 @@ jobs:
         shell: bash -l {0}
         run: |
           export PATH=/usr/share/miniconda/bin:$PATH
-          python setup.py test
+          pytest -n 2
 
       - name: Save Test Results
         if: always()

develop.txt

@@ -1,7 +1,11 @@
 coverage>=5.5
 pytest>=6.2.2
+pytest-raises>=0.10
+pytest-cases>= 3.6
+pytest-xdist>= 3.0.1
 pytest-cov>=2.11.1
 pytest-emoji>=0.2.0
+pydocstyle==6.1.1
 pytest-pydocstyle>=2.2.0
 black
 isort

modopt/opt/linear.py

@@ -11,6 +11,7 @@
 import numpy as np
 
 from modopt.base.types import check_callable, check_float
+from modopt.base.backend import get_array_module
 from modopt.signal.wavelet import filter_convolve_stack
 
 
@@ -80,6 +81,23 @@ def __init__(self):
         self.adj_op = self.op
 
 
+class MatrixOperator(LinearParent):
+    """
+    Matrix Operator class.
+
+    This class transforms an array into a suitable linear operator.
+    """
+
+    def __init__(self, array):
+        self.op = lambda x: array @ x
+        xp = get_array_module(array)
+
+        if xp.any(xp.iscomplex(array)):
+            self.adj_op = lambda x: array.T.conjugate() @ x
+        else:
+            self.adj_op = lambda x: array.T @ x
+
+
 class WaveletConvolve(LinearParent):
     """Wavelet Convolution Class.