Skip to content

Allow sequences for center in CalculusPatch #267

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
inducer merged 1 commit into from
Mar 15, 2026
Merged

Allow sequences for center in CalculusPatch #267

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 0 additions & 16 deletions .basedpyright/baseline.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13687,22 +13687,6 @@
"lineCount": 1
}
},
{
"code": "reportUnknownMemberType",
"range": {
"startColumn": 24,
"endColumn": 50,
"lineCount": 1
}
},
{
"code": "reportUnknownMemberType",
"range": {
"startColumn": 25,
"endColumn": 52,
"lineCount": 1
}
},
{
"code": "reportUnknownArgumentType",
"range": {
Expand Down
1 change: 1 addition & 0 deletions doc/conf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@
"Array1D": "class:numpy.ndarray",
"Array2D": "class:numpy.ndarray",
"ArrayND": "class:numpy.ndarray",
"ToArray1D": "class:numpy.ndarray",
"np.floating": "class:numpy.floating",
"np.complexfloating": "class:numpy.complexfloating",
"np.inexact": "class:numpy.inexact",
Expand Down
18 changes: 9 additions & 9 deletions sumpy/point_calculus.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@
if TYPE_CHECKING:
from collections.abc import Callable, Sequence

from optype.numpy import Array1D, Array2D, ArrayND
from optype.numpy import Array1D, Array2D, ArrayND, ToArray1D


__doc__ = """
Expand Down Expand Up @@ -99,28 +99,28 @@ class CalculusPatch:
_pshape: tuple[int, ...]

def __init__(self,
center: Array1D[np.floating[Any]],
center: ToArray1D[np.floating[Any]],
h: float = 1e-1,
order: int = 4,
nodes: NodesKind = "chebyshev") -> None:
self.center = center
dtype = center.dtype
center = np.asarray(center)
assert center.ndim == 1

npoints = order + 1
if nodes == "equispaced":
points_1d = np.linspace(-h/2, h/2, npoints, dtype=dtype)
points_1d = np.linspace(-h/2, h/2, npoints)
weights_1d = None

elif nodes == "chebyshev":
a = np.arange(npoints, dtype=dtype)
a = np.arange(npoints)
points_1d = (h/2)*np.cos((2*(a+1)-1)/(2*npoints)*np.pi)
weights_1d = None

elif nodes == "legendre":
from scipy.special import legendre
points_1d, weights_1d, _ = legendre(npoints).weights.T
points_1d = (points_1d * (h/2)).astype(dtype)
weights_1d = (weights_1d * (h/2)).astype(dtype)
points_1d = points_1d * (h/2)
weights_1d = weights_1d * (h/2)

else:
raise ValueError(f"invalid node set: {nodes}")
Expand All @@ -130,8 +130,8 @@ def __init__(self,
self._points_1d = points_1d
self._weights_1d = weights_1d

self.dim = dim = len(self.center)
self.center = center
self.dim = dim = len(self.center)

points_shaped = np.array(np.meshgrid(
*[center[i] + points_1d for i in range(dim)],
Expand Down
Loading