split public api into 1d,2d,nd and machine learning test functions

SimonBlanke · Jan 1, 2024 · abba8fb · abba8fb
1 parent 0c5c645
commit abba8fb
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 1 deletion.
diff --git a/surfaces/__init__.py b/surfaces/__init__.py
@@ -5,11 +5,20 @@
 __version__ = "0.1.0"
 __license__ = "MIT"
 
-from .mathematical_functions import mathematical_functions
+from .mathematical_functions import (
+    mathematical_functions,
+    mathematical_functions_1d,
+    mathematical_functions_2d,
+    mathematical_functions_nd,
+)
+
 from .machine_learning_functions import machine_learning_functions
 
 
 __all__ = [
     "mathematical_functions",
+    "mathematical_functions_1d",
+    "mathematical_functions_2d",
+    "mathematical_functions_nd",
     "machine_learning_functions",
 ]
diff --git a/surfaces/mathematical_functions/__init__.py b/surfaces/mathematical_functions/__init__.py
@@ -81,3 +81,39 @@
     StyblinskiTangFunction,
     ThreeHumpCamelFunction,
 ]
+
+
+mathematical_functions_1d = [
+    GramacyAndLeeFunction,
+]
+
+
+mathematical_functions_2d = [
+    AckleyFunction,
+    BealeFunction,
+    BoothFunction,
+    BukinFunctionN6,
+    CrossInTrayFunction,
+    DropWaveFunction,
+    EasomFunction,
+    EggholderFunction,
+    GoldsteinPriceFunction,
+    HimmelblausFunction,
+    HölderTableFunction,
+    LangermannFunction,
+    LeviFunctionN13,
+    MatyasFunction,
+    McCormickFunction,
+    RosenbrockFunction,
+    SchafferFunctionN2,
+    SimionescuFunction,
+    ThreeHumpCamelFunction,
+]
+
+
+mathematical_functions_nd = [
+    GriewankFunction,
+    RastriginFunction,
+    SphereFunction,
+    StyblinskiTangFunction,
+]