1010
1111import numpy as np
1212
13- from csaps ._base import SplinePPFormBase , ISmoothingSpline
14- from csaps ._types import UnivariateDataType , NdGridDataType
15- from csaps ._sspumv import SplinePPForm , UnivariateCubicSmoothingSpline
13+ from ._base import SplinePPFormBase , ISmoothingSpline
14+ from ._types import UnivariateDataType , NdGridDataType
15+ from ._sspumv import SplinePPForm , CubicSmoothingSpline
1616
1717
1818def ndgrid_prepare_data_sites (data , name ) -> ty .Tuple [np .ndarray , ...]:
1919 if not isinstance (data , c_abc .Sequence ):
20- raise TypeError ("'{}' must be a sequence of the vectors." . format ( name ) )
20+ raise TypeError (f "'{ name } ' must be a sequence of the vectors." )
2121
2222 data = list (data )
2323
2424 for i , di in enumerate (data ):
2525 di = np .array (di , dtype = np .float64 )
2626 if di .ndim > 1 :
27- raise ValueError ("All '{}' elements must be a vector." . format ( name ) )
27+ raise ValueError (f "All '{ name } ' elements must be a vector." )
2828 if di .size < 2 :
29- raise ValueError (
30- "'{}' must contain at least 2 data points." .format (name ))
29+ raise ValueError (f"'{ name } ' must contain at least 2 data points." )
3130 data [i ] = di
3231
3332 return tuple (data )
@@ -165,28 +164,24 @@ def _prepare_data(cls, xdata, ydata, weights, smooth):
165164 data_ndim = len (xdata )
166165
167166 if ydata .ndim != data_ndim :
168- raise ValueError (
169- 'ydata must have dimension {} according to xdata' .format (data_ndim ))
167+ raise ValueError (f'ydata must have dimension { data_ndim } according to xdata' )
170168
171169 for yd , xs in zip (ydata .shape , map (len , xdata )):
172170 if yd != xs :
173- raise ValueError (
174- 'ydata ({}) and xdata ({}) dimension size mismatch' .format (yd , xs ))
171+ raise ValueError (f'ydata ({ yd } ) and xdata ({ xs } ) dimension size mismatch' )
175172
176173 if not weights :
177174 weights = [None ] * data_ndim
178175 else :
179176 weights = ndgrid_prepare_data_sites (weights , 'weights' )
180177
181178 if len (weights ) != data_ndim :
182- raise ValueError (
183- 'weights ({}) and xdata ({}) dimensions mismatch' .format (len (weights ), data_ndim ))
179+ raise ValueError (f'weights ({ len (weights )} ) and xdata ({ data_ndim } ) dimensions mismatch' )
184180
185181 for w , x in zip (weights , xdata ):
186182 if w is not None :
187183 if w .size != x .size :
188- raise ValueError (
189- 'weights ({}) and xdata ({}) dimension size mismatch' .format (w , x ))
184+ raise ValueError (f'weights ({ w } ) and xdata ({ x } ) dimension size mismatch' )
190185
191186 if not smooth :
192187 smooth = [None ] * data_ndim
@@ -198,8 +193,8 @@ def _prepare_data(cls, xdata, ydata, weights, smooth):
198193
199194 if len (smooth ) != data_ndim :
200195 raise ValueError (
201- 'Number of smoothing parameter values must be equal '
202- ' number of dimensions ({})'. format ( data_ndim ) )
196+ f 'Number of smoothing parameter values must '
197+ f'be equal number of dimensions ({ data_ndim } )' )
203198
204199 return xdata , ydata , weights , smooth
205200
@@ -208,9 +203,8 @@ def __call__(self, xi: NdGridDataType) -> np.ndarray:
208203 """
209204 xi = ndgrid_prepare_data_sites (xi , 'xi' )
210205
211- if len (xi ) != self ._ndim :
212- raise ValueError (
213- 'xi ({}) and xdata ({}) dimensions mismatch' .format (len (xi ), self ._ndim ))
206+ if len (xi ) != self ._ndim : # pragma: no cover
207+ raise ValueError (f'xi ({ len (xi )} ) and xdata ({ self ._ndim } ) dimensions mismatch' )
214208
215209 return self ._spline .evaluate (xi )
216210
@@ -224,8 +218,8 @@ def _make_spline(self, smooth: ty.List[ty.Optional[float]]) -> ty.Tuple[NdGridSp
224218 shape_i = (np .prod (sizey [:- 1 ]), sizey [- 1 ])
225219 ydata_i = ydata .reshape (shape_i , order = 'F' )
226220
227- s = UnivariateCubicSmoothingSpline (
228- self ._xdata [i ], ydata_i , self ._weights [i ], smooth [i ])
221+ s = CubicSmoothingSpline (
222+ self ._xdata [i ], ydata_i , weights = self ._weights [i ], smooth = smooth [i ])
229223
230224 _smooth .append (s .smooth )
231225 sizey [- 1 ] = s .spline .pieces * s .spline .order
0 commit comments