fix imports in multiple tests

SimonBlanke · Apr 20, 2024 · b9417de · b9417de
1 parent d82f846
commit b9417de
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 66 deletions.
diff --git a/tests/test_all_test_functions.py b/tests/test_all_test_functions.py
@@ -3,7 +3,7 @@
 from hyperactive import Hyperactive
 from gradient_free_optimizers import RandomSearchOptimizer
 
-from surfaces import mathematical_functions, machine_learning_functions
+from surfaces.test_functions import mathematical_functions, machine_learning_functions
 
 
 mathematical_functions_d = (
@@ -27,7 +27,7 @@ def test_all_mathematical_functions(test_function):
 
     objective_function = test_function_.objective_function
     search_space = test_function_.search_space(value_types="array")
-    n_iter = 20
+    n_iter = 100
 
     opt = RandomSearchOptimizer(search_space)
     opt.search(objective_function, n_iter=n_iter)
@@ -39,7 +39,7 @@ def test_all_machine_learning_functions(test_function):
 
     objective_function = test_function_.objective_function
     search_space = test_function_.search_space()
-    n_iter = 20
+    n_iter = 3
 
     hyper = Hyperactive()
     hyper.add_search(

diff --git a/tests/test_arrays_input.py b/tests/test_arrays_input.py
diff --git a/tests/test_optimization.py b/tests/test_optimization.py
@@ -2,7 +2,7 @@
 import numpy as np
 from gradient_free_optimizers import RandomSearchOptimizer
 
-from surfaces.mathematical_functions import (
+from surfaces.test_functions.mathematical import (
     SphereFunction,
     RastriginFunction,
     AckleyFunction,