diff --git a/utils/chebutils.py b/utils/chebutils.py index 0d7e38a..cf47567 100644 --- a/utils/chebutils.py +++ b/utils/chebutils.py @@ -2,8 +2,19 @@ def chebD(n): """ - Hint: https://people.maths.ox.ac.uk/trefethen/book.pdf, Chapter 6, pages - 51-55. + Computes a spectral differentiation matrix and Chebysev grid + + Parameters + ---------- + n: + An integer defining the number of grid points (n+1) + + Returns + ------- + D: (n+1)x(n+1) matrix + differentiation matrix + x: (n+1)-d array + Chebysev grid """ if n == 0: x = 1; D = 0; w = 0 diff --git a/utils/quadroots.py b/utils/quadroots.py index 460f2b1..4cb77ed 100644 --- a/utils/quadroots.py +++ b/utils/quadroots.py @@ -21,7 +21,7 @@ def quadroots(a, b, c): Roots of the equation. """ - x1 = (-b + np.sqrt(b**2 - 4*a*c))/2*a - x2 = (-b - np.sqrt(b**2 - 4*a*c))/2*a + x1 = (-b + np.sqrt(b**2 - 4*a*c))/(2*a) + x2 = (-b - np.sqrt(b**2 - 4*a*c))/(2*a) return (x1, x2)