Skip to content

Commit

Permalink
finished testing and updated example notebooks; adding test for Trimm…
Browse files Browse the repository at this point in the history
…edLibrary; merged update from master
  • Loading branch information
znicolaou committed Dec 2, 2023
1 parent cb8b3df commit b71e7f7
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions test/test_feature_library.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@
from pysindy import SINDy
from pysindy.differentiation import FiniteDifference
from pysindy.feature_library import ConcatLibrary
from pysindy.feature_library import TrimmedLibrary
from pysindy.feature_library import CustomLibrary
from pysindy.feature_library import FourierLibrary
from pysindy.feature_library import GeneralizedLibrary
Expand All @@ -20,6 +19,7 @@
from pysindy.feature_library import PolynomialLibrary
from pysindy.feature_library import SINDyPILibrary
from pysindy.feature_library import TensoredLibrary
from pysindy.feature_library import TrimmedLibrary
from pysindy.feature_library import WeakPDELibrary
from pysindy.feature_library.base import BaseFeatureLibrary
from pysindy.optimizers import SINDyPI
Expand Down Expand Up @@ -419,18 +419,20 @@ def test_tensored(data_lorenz):
tensored_lib.fit_transform(x)
check_is_fitted(tensored_lib)


def test_trimmed(data_lorenz):
x, t = data_lorenz
ident_lib = IdentityLibrary()
poly_lib = PolynomialLibrary()
tensored_lib = ident_lib * poly_lib
trimmed_lib = TrimmedLibrary(tensored_lib, drop_inds=[0,1])
trimmed_lib = TrimmedLibrary(tensored_lib, drop_inds=[0, 1])
assert isinstance(trimmed_lib, TrimmedLibrary)
trimmed_lib.fit(x)
check_is_fitted(trimmed_lib)
trimmed_lib.fit_transform(x)
check_is_fitted(trimmed_lib)


@pytest.mark.parametrize(
"library",
[
Expand Down

0 comments on commit b71e7f7

Please sign in to comment.