Replace roots module with scipy.special functions

labrosse · Feb 25, 2023 · 19c9373 · 19c9373
1 parent a65875e
commit 19c9373
Show file tree

Hide file tree

Showing 8 changed files with 3 additions and 78 deletions.
diff --git a/dmsuite/poly_diff.py b/dmsuite/poly_diff.py
@@ -28,8 +28,7 @@
 import numpy as np
 from numpy.typing import NDArray
 from scipy.linalg import toeplitz
-
-from .roots import herroots, lagroots
+from scipy.special import roots_hermite, roots_laguerre
 
 
 class DiffMatrices(ABC):
@@ -255,7 +254,7 @@ def max_order(self) -> int:
 
  @cached_property
  def nodes(self) -> NDArray:
- return herroots(self.degree)
+ return roots_hermite(self.degree)[0]
 
  @cached_property
  def _dmat(self) -> GeneralPoly:
@@ -296,7 +295,7 @@ def max_order(self) -> int:
  @cached_property
  def nodes(self) -> NDArray:
  nodes = np.zeros(self.degree + 1)
- nodes[1:] = lagroots(self.degree)
+ nodes[1:] = roots_laguerre(self.degree)[0]
  return nodes
 
  @cached_property

diff --git a/dmsuite/roots.py b/dmsuite/roots.py
diff --git a/tests/data/herroots10.npy b/tests/data/herroots10.npy
diff --git a/tests/data/lagroots10.npy b/tests/data/lagroots10.npy
diff --git a/tests/data/legroots10.npy b/tests/data/legroots10.npy
diff --git a/tests/test_herroots.py b/tests/test_herroots.py
diff --git a/tests/test_lagroots.py b/tests/test_lagroots.py
diff --git a/tests/test_legroots.py b/tests/test_legroots.py