From bb97b60376f17a917779c71867b561681fa71289 Mon Sep 17 00:00:00 2001
From: Thomas <purcell@fhi-berlin.mpg.de>
Date: Wed, 6 Jan 2021 10:47:30 +0100
Subject: [PATCH] Update tests to match new top level format

still need to add tests for this
---
 .../model/test_model_log_regressor_fit_c.cc                     | 2 +-
 .../sisso_regressor/test_sisso_log_regressor_fit_c.cc           | 2 +-
 .../feature_creation/parameterization/test_abs_diff_node.cc     | 2 +-
 .../feature_creation/parameterization/test_abs_node.cc          | 2 +-
 .../feature_creation/parameterization/test_add_node.cc          | 2 +-
 .../feature_creation/parameterization/test_cb_node.cc           | 2 +-
 .../feature_creation/parameterization/test_cbrt_node.cc         | 2 +-
 .../feature_creation/parameterization/test_cos_node.cc          | 2 +-
 .../feature_creation/parameterization/test_div_node.cc          | 2 +-
 .../feature_creation/parameterization/test_exp_node.cc          | 2 +-
 .../feature_creation/parameterization/test_inv_node.cc          | 2 +-
 .../feature_creation/parameterization/test_log_node.cc          | 2 +-
 .../feature_creation/parameterization/test_mult_node.cc         | 2 +-
 .../feature_creation/parameterization/test_neg_exp_node.cc      | 2 +-
 .../feature_creation/parameterization/test_sin_node.cc          | 2 +-
 .../feature_creation/parameterization/test_six_pow_node.cc      | 2 +-
 .../feature_creation/parameterization/test_sq_node.cc           | 2 +-
 .../feature_creation/parameterization/test_sqrt_node.cc         | 2 +-
 .../feature_creation/parameterization/test_sub_node.cc          | 2 +-
 .../test_descriptor_identifier/test_log_regressor_fit_c.py      | 2 +-
 .../test_feature_creation/test_parameterize/test_lorentizan.py  | 2 +-
 .../test_feature_creation/test_parameterize/test_param_abs.py   | 2 +-
 .../test_parameterize/test_param_abs_diff.py                    | 2 +-
 .../test_feature_creation/test_parameterize/test_param_add.py   | 2 +-
 .../test_feature_creation/test_parameterize/test_param_cb.py    | 2 +-
 .../test_feature_creation/test_parameterize/test_param_cbrt.py  | 2 +-
 .../test_feature_creation/test_parameterize/test_param_cos.py   | 2 +-
 .../test_feature_creation/test_parameterize/test_param_div.py   | 2 +-
 .../test_feature_creation/test_parameterize/test_param_exp.py   | 2 +-
 .../test_feature_creation/test_parameterize/test_param_inv.py   | 2 +-
 .../test_feature_creation/test_parameterize/test_param_log.py   | 2 +-
 .../test_parameterize/test_param_neg_exp.py                     | 2 +-
 .../test_feature_creation/test_parameterize/test_param_sin.py   | 2 +-
 .../test_parameterize/test_param_six_pow.py                     | 2 +-
 .../test_feature_creation/test_parameterize/test_param_sq.py    | 2 +-
 .../test_feature_creation/test_parameterize/test_param_sqrt.py  | 2 +-
 .../test_feature_creation/test_parameterize/test_param_sub.py   | 2 +-
 37 files changed, 37 insertions(+), 37 deletions(-)

diff --git a/tests/googletest/descriptor_identification/model/test_model_log_regressor_fit_c.cc b/tests/googletest/descriptor_identification/model/test_model_log_regressor_fit_c.cc
index d4abfa57..a36f5dff 100644
--- a/tests/googletest/descriptor_identification/model/test_model_log_regressor_fit_c.cc
+++ b/tests/googletest/descriptor_identification/model/test_model_log_regressor_fit_c.cc
@@ -12,7 +12,7 @@ namespace
             _task_sizes_train = {10};
             _task_sizes_test = {2};
 
-            nlopt_wrapper::setup_data(_task_sizes_train, 2, 0);
+            nlopt_wrapper::setup_data(_task_sizes_train, 2, 0, false);
 
             std::vector<double> value_1 = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0};
             std::vector<double> test_value_1 = {1.0, 7.0};
diff --git a/tests/googletest/descriptor_identification/sisso_regressor/test_sisso_log_regressor_fit_c.cc b/tests/googletest/descriptor_identification/sisso_regressor/test_sisso_log_regressor_fit_c.cc
index b15f8169..4ee0b3ac 100644
--- a/tests/googletest/descriptor_identification/sisso_regressor/test_sisso_log_regressor_fit_c.cc
+++ b/tests/googletest/descriptor_identification/sisso_regressor/test_sisso_log_regressor_fit_c.cc
@@ -19,7 +19,7 @@ namespace
             _task_sizes_train = {90};
             _task_sizes_test = {10};
             _leave_out_inds = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
-            nlopt_wrapper::setup_data(_task_sizes_train, 2, 0);
+            nlopt_wrapper::setup_data(_task_sizes_train, 2, 0, false);
 
             std::vector<double> value_1(90, 0.0);
             std::vector<double> value_2(90, 0.0);
diff --git a/tests/googletest/feature_creation/parameterization/test_abs_diff_node.cc b/tests/googletest/feature_creation/parameterization/test_abs_diff_node.cc
index df9d06f0..f2f32940 100644
--- a/tests/googletest/feature_creation/parameterization/test_abs_diff_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_abs_diff_node.cc
@@ -48,7 +48,7 @@ namespace
 
             allowed_op_funcs::abs_diff(90, _phi[0]->value_ptr(), _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_abs_node.cc b/tests/googletest/feature_creation/parameterization/test_abs_node.cc
index f0ba7bfc..c5113aaa 100644
--- a/tests/googletest/feature_creation/parameterization/test_abs_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_abs_node.cc
@@ -38,7 +38,7 @@ namespace
 
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::abs(900, _phi[0]->value_ptr(), _alpha, _a, _prop.data());
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_add_node.cc b/tests/googletest/feature_creation/parameterization/test_add_node.cc
index 2f2b9c34..b149c0a2 100644
--- a/tests/googletest/feature_creation/parameterization/test_add_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_add_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::add(90, _phi[0]->value_ptr(), _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_cb_node.cc b/tests/googletest/feature_creation/parameterization/test_cb_node.cc
index ada3cc8e..e505f2fb 100644
--- a/tests/googletest/feature_creation/parameterization/test_cb_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_cb_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::cb(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_cbrt_node.cc b/tests/googletest/feature_creation/parameterization/test_cbrt_node.cc
index 57e0055a..7699bee3 100644
--- a/tests/googletest/feature_creation/parameterization/test_cbrt_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_cbrt_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::cbrt(900, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_cos_node.cc b/tests/googletest/feature_creation/parameterization/test_cos_node.cc
index 9f63b56e..f47bed4d 100644
--- a/tests/googletest/feature_creation/parameterization/test_cos_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_cos_node.cc
@@ -52,7 +52,7 @@ namespace
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::cos(900, _phi[0]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_div_node.cc b/tests/googletest/feature_creation/parameterization/test_div_node.cc
index 2cfc6b3d..0f0404f6 100644
--- a/tests/googletest/feature_creation/parameterization/test_div_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_div_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::div(90, _phi[0]->value_ptr(), _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_exp_node.cc b/tests/googletest/feature_creation/parameterization/test_exp_node.cc
index 7b7b0751..959e79ee 100644
--- a/tests/googletest/feature_creation/parameterization/test_exp_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_exp_node.cc
@@ -55,7 +55,7 @@ namespace
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::exp(900, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_inv_node.cc b/tests/googletest/feature_creation/parameterization/test_inv_node.cc
index dbf825ea..0d8cad18 100644
--- a/tests/googletest/feature_creation/parameterization/test_inv_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_inv_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::inv(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_log_node.cc b/tests/googletest/feature_creation/parameterization/test_log_node.cc
index 81e3cb5b..def50c0d 100644
--- a/tests/googletest/feature_creation/parameterization/test_log_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_log_node.cc
@@ -55,7 +55,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::log(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_mult_node.cc b/tests/googletest/feature_creation/parameterization/test_mult_node.cc
index 11004cea..8db88335 100644
--- a/tests/googletest/feature_creation/parameterization/test_mult_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_mult_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::mult(900, _phi[0]->value_ptr(), _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_neg_exp_node.cc b/tests/googletest/feature_creation/parameterization/test_neg_exp_node.cc
index 4b1b471c..3aca1341 100644
--- a/tests/googletest/feature_creation/parameterization/test_neg_exp_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_neg_exp_node.cc
@@ -55,7 +55,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::neg_exp(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_sin_node.cc b/tests/googletest/feature_creation/parameterization/test_sin_node.cc
index 796e4c21..4f2418d8 100644
--- a/tests/googletest/feature_creation/parameterization/test_sin_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_sin_node.cc
@@ -52,7 +52,7 @@ namespace
             _prop = std::vector<double>(900, 0.0);
             allowed_op_funcs::sin(900, _phi[0]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_six_pow_node.cc b/tests/googletest/feature_creation/parameterization/test_six_pow_node.cc
index 424e73a9..f1e3c20f 100644
--- a/tests/googletest/feature_creation/parameterization/test_six_pow_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_six_pow_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::sixth_pow(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_sq_node.cc b/tests/googletest/feature_creation/parameterization/test_sq_node.cc
index 4b773d49..fe80642e 100644
--- a/tests/googletest/feature_creation/parameterization/test_sq_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_sq_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::sq(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_sqrt_node.cc b/tests/googletest/feature_creation/parameterization/test_sqrt_node.cc
index 938ab6e4..bc560ea1 100644
--- a/tests/googletest/feature_creation/parameterization/test_sqrt_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_sqrt_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::sqrt(90, _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/googletest/feature_creation/parameterization/test_sub_node.cc b/tests/googletest/feature_creation/parameterization/test_sub_node.cc
index 2a112b60..7582e533 100644
--- a/tests/googletest/feature_creation/parameterization/test_sub_node.cc
+++ b/tests/googletest/feature_creation/parameterization/test_sub_node.cc
@@ -47,7 +47,7 @@ namespace
             _prop = std::vector<double>(90, 0.0);
             allowed_op_funcs::sub(90, _phi[0]->value_ptr(), _phi[1]->value_ptr(), _alpha, _a, _prop.data());
 
-            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1);
+            nlopt_wrapper::set_objective("regression", _prop.data(), _task_sizes_train, 1, true);
         }
 
         node_ptr _feat_1;
diff --git a/tests/pytest/test_descriptor_identifier/test_log_regressor_fit_c.py b/tests/pytest/test_descriptor_identifier/test_log_regressor_fit_c.py
index 593077ab..a78c90dd 100644
--- a/tests/pytest/test_descriptor_identifier/test_log_regressor_fit_c.py
+++ b/tests/pytest/test_descriptor_identifier/test_log_regressor_fit_c.py
@@ -36,7 +36,7 @@ def test_sisso_log_regressor_fit_c():
 
     op_set = ["add", "sub", "mult", "sq", "cb", "sqrt", "cbrt"]
 
-    setup_data([90], 2, 0)
+    setup_data([90], 2, 0, True)
     feat_space = generate_fs(
         phi_0, prop, [90], op_set, [], "log_regression_fit_c", 0, 5
     )
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_lorentizan.py b/tests/pytest/test_feature_creation/test_parameterize/test_lorentizan.py
index 2700d3bd..f9ec0ca4 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_lorentizan.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_lorentizan.py
@@ -30,7 +30,7 @@ def test_lorentzian():
     feat_node = InvParamNode(SqNode(feat_1, 2, 1e-50, 1e50), 3, 1e-50, 1e50)
     prop = 21.4 / ((data_1 - 0.25) ** 2.0 + 7.1) - 1.478
 
-    set_objective("regression", prop, [900], 2)
+    set_objective("regression", prop, [900], 2, True)
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
 
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_abs.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_abs.py
index 5dd45aa6..698af8e4 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_abs.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_abs.py
@@ -31,7 +31,7 @@ def test_param_abs_node():
 
     prop = -2.3 * np.abs(1.55 * data_1 + 0.8751) - 1.2
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_abs_diff.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_abs_diff.py
index 45e8106e..e2cdd0fe 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_abs_diff.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_abs_diff.py
@@ -34,7 +34,7 @@ def test_param_abs_diff_node():
     feat_node = AbsDiffParamNode(feat_1, feat_2, 2, 1e-50, 1e50)
 
     prop = -2.3 * np.abs(data_1 - (1.5 * data_2 + 0.8751)) - 1.2
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_add.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_add.py
index 28983c8b..de5d697d 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_add.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_add.py
@@ -35,7 +35,7 @@ def test_param_add_node():
 
     prop = -2.3 * (data_1 + 1.5 * data_2) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_cb.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_cb.py
index 979c3a86..49344208 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_cb.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_cb.py
@@ -31,7 +31,7 @@ def test_param_cb_node():
 
     prop = np.power(1.55 * data_1 + 0.8751, 3.0) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_cbrt.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_cbrt.py
index 4d8d74bc..0e3f672f 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_cbrt.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_cbrt.py
@@ -31,7 +31,7 @@ def test_param_cbrt_node():
 
     prop = np.cbrt(1.55 * data_1 + 0.8751) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_cos.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_cos.py
index 4669c7ed..86ca7519 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_cos.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_cos.py
@@ -31,7 +31,7 @@ def test_param_cos_node():
 
     prop = -1.1 * np.cos(1.25 * data_1 + 2.13) + 0.01578
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
 
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_div.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_div.py
index 3440ba24..d937bf04 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_div.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_div.py
@@ -36,7 +36,7 @@ def test_param_div_node():
 
     prop = 4.124 * data_1 / ((data_2 + 1.8751)) - 0.12
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
 
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_exp.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_exp.py
index bd0b141f..1eda227b 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_exp.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_exp.py
@@ -31,7 +31,7 @@ def test_param_exp_node():
 
     prop = np.exp(1.32 * data_1 + 0.8751) - 0.12
 
-    set_objective("regression", prop, [900], 10)
+    set_objective("regression", prop, [900], 10, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_inv.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_inv.py
index 253fd9d9..67a3617e 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_inv.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_inv.py
@@ -32,7 +32,7 @@ def test_param_inv_node():
 
     prop = 1.0 / (1.55 * data_1 + 0.8751) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_log.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_log.py
index 4499f824..5458d29d 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_log.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_log.py
@@ -31,7 +31,7 @@ def test_param_log_node():
 
     prop = -2.014 * np.log(1.15 * data_1 + 0.1387)
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_neg_exp.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_neg_exp.py
index 71cf89e2..7a91c74c 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_neg_exp.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_neg_exp.py
@@ -31,7 +31,7 @@ def test_param_neg_exp_node():
 
     prop = np.exp(-1.55 * data_1 + 0.8751) - 0.12
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_sin.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_sin.py
index 5424a432..c1ba0c0c 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_sin.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_sin.py
@@ -31,7 +31,7 @@ def test_param_sin_node():
 
     prop = -1.1 * np.sin(1.25 * data_1 + 2.13) + 0.01578
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
 
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_six_pow.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_six_pow.py
index 950d2172..bbdd4db1 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_six_pow.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_six_pow.py
@@ -31,7 +31,7 @@ def test_param_six_pow_node():
 
     prop = 1.55 * np.power(data_1 + 0.21, 6.0) - 0.12
 
-    set_objective("regression", prop, [900], 1)
+    set_objective("regression", prop, [900], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_sq.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_sq.py
index 4d54a05a..02a08594 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_sq.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_sq.py
@@ -31,7 +31,7 @@ def test_param_sq_node():
 
     prop = np.power(1.55 * data_1 + 0.8751, 2.0) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_sqrt.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_sqrt.py
index 2ffc981a..6e410d63 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_sqrt.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_sqrt.py
@@ -31,7 +31,7 @@ def test_param_sqrt_node():
 
     prop = np.sqrt(1.55 * data_1 + 0.8751) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     assert check_feat_parmeters(feat_node, prop)
diff --git a/tests/pytest/test_feature_creation/test_parameterize/test_param_sub.py b/tests/pytest/test_feature_creation/test_parameterize/test_param_sub.py
index 7de4ab9b..11dbbc10 100644
--- a/tests/pytest/test_feature_creation/test_parameterize/test_param_sub.py
+++ b/tests/pytest/test_feature_creation/test_parameterize/test_param_sub.py
@@ -35,7 +35,7 @@ def test_param_sub_node():
 
     prop = -2.3 * (data_1 - 1.5 * data_2) - 1.2
 
-    set_objective("regression", prop, [90], 1)
+    set_objective("regression", prop, [90], 1, True)
 
     feat_node.get_parameters(prop)
     check_feat_parmeters(feat_node, prop)
-- 
GitLab