10000 Fix some triangulation types · matplotlib/matplotlib@736acc6 · GitHub
[go: up one dir, main page]
Skip to content

Commit 736acc6

Browse files
QuLogicQuLogic
committed
Fix some triangulation types
1 parent 3e95c36 commit 736acc6

File tree

3 files changed

+18
-3
lines changed

3 files changed

+18
-3
lines changed

‎lib/matplotlib/tri/_trifinder.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,9 @@ def __init__(self, triangulation):
2020
_api.check_isinstance(Triangulation, triangulation=triangulation)
2121
self._triangulation = triangulation
2222

23+
def __call__(self, x, y):
24+
raise NotImplementedError
25+
2326

2427
class TrapezoidMapTriFinder(TriFinder):
2528
"""

‎lib/matplotlib/tri/_trifinder.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ from numpy.typing import ArrayLike
33

44
class TriFinder:
55
def __init__(self, triangulation: Triangulation) -> None: ...
6+
def __call__(self, x: ArrayLike, y: ArrayLike) -> ArrayLike: ...
67

78
class TrapezoidMapTriFinder(TriFinder):
89
def __init__(self, triangulation: Triangulation) -> None: ...

‎lib/matplotlib/tri/_trirefine.pyi

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,28 @@
1-
from matplotlib.tri._triangulation import Triangulation
2-
from matplotlib.tri._triinterpolate import TriInterpolator
1+
from typing import Literal, overload
32

43
import numpy as np
54
from numpy.typing import ArrayLike
65

6+
from matplotlib.tri._triangulation import Triangulation
7+
from matplotlib.tri._triinterpolate import TriInterpolator
8+
79
class TriRefiner:
810
def __init__(self, triangulation: Triangulation) -> None: ...
911

1012
class UniformTriRefiner(TriRefiner):
1113
def __init__(self, triangulation: Triangulation) -> None: ...
14+
@overload
1215
def refine_triangulation(
13-
self, return_tri_index: bool = ..., subdiv: int = ...
16+
self, *, return_tri_index: Literal[True], subdiv: int = ...
1417
) -> tuple[Triangulation, np.ndarray]: ...
18+
@overload
19+
def refine_triangulation(
20+
self, return_tri_index: Literal[False] = ..., subdiv: int = ...
21+
) -> Triangulation: ...
22+
@overload
23+
def refine_triangulation(
24+
self, return_tri_index: bool = ..., subdiv: int = ...
25+
) -> tuple[Triangulation, np.ndarray] | Triangulation: ...
1526
def refine_field(
1627
self,
1728
z: ArrayLike,

0 commit comments

Comments
 (0)
0