Add multivariable function to custom library tests

dynamicslab · Sep 26, 2020 · 0f7ca74 · 0f7ca74
1 parent 6ece429
commit 0f7ca74
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -117,8 +117,18 @@ def data_derivative_2d():
 
 @pytest.fixture
 def data_custom_library():
-    library_functions = [lambda x: x, lambda x: x ** 2, lambda x: 0 * x]
-    function_names = [lambda s: str(s), lambda s: str(s) + "^2", lambda s: "0"]
+    library_functions = [
+        lambda x: x,
+        lambda x: x ** 2,
+        lambda x: 0 * x,
+        lambda x, y: x * y,
+    ]
+    function_names = [
+        lambda s: str(s),
+        lambda s: str(s) + "^2",
+        lambda s: "0",
+        lambda s, t: str(s) + " " + str(t),
+    ]
 
     return CustomLibrary(
         library_functions=library_functions, function_names=function_names

diff --git a/test/feature_library/test_feature_library.py b/test/feature_library/test_feature_library.py
@@ -94,7 +94,7 @@ def test_change_in_data_shape(data_lorenz, library):
         (PolynomialLibrary(), 10),
         (IdentityLibrary() + PolynomialLibrary(), 13),
         (FourierLibrary(), 6),
-        (pytest.lazy_fixture("data_custom_library"), 9),
+        (pytest.lazy_fixture("data_custom_library"), 12),
     ],
 )
 def test_output_shape(data_lorenz, library, shape):