Skip to content
Snippets Groups Projects
Commit aa0848af authored by Thomas Purcell's avatar Thomas Purcell
Browse files

Fix the pytests for the parameterization scheme

These tests needed to be fixed for the updated scheme
parent 2c5d7c9e
No related branches found
No related tags found
No related merge requests found
......@@ -32,7 +32,7 @@ def test_lorentzian():
feat_node = InvParamNode(SqNode(feat_1, 2, 1e-50, 1e50), 3, 1e-50, 1e50)
feat_node.get_parameters(optimizer)
print(feat_node.parameters)
assert check_feat_parmeters(feat_node, prop)
......
......@@ -20,14 +20,14 @@ def check_feat_parmeters(feat, prop):
def test_param_log_node():
initialize_values_arr(90, 10, 1, 1)
initialize_values_arr(900, 10, 1, 1)
data_1 = np.linspace(0.5, 20, 90)
data_1 = np.linspace(0.5, 20, 900)
test_data_1 = np.linspace(0.52145, 19.99, 10)
feat_1 = FeatureNode(0, "t_a", data_1, test_data_1, Unit())
prop = -2.014 * np.log(1.15 * data_1 + 0.1387)
optimizer = get_reg_optimizer([90], prop, 1, 1, 0.5)
optimizer = get_reg_optimizer([900], prop, 1, 1, 0.5)
feat_node = LogParamNode(feat_1, 2, 1e-50, 1e50)
feat_node.get_parameters(optimizer)
......
......@@ -25,8 +25,8 @@ def test_param():
shutil.rmtree("models/")
shutil.rmtree("feature_space/")
assert sisso.models[0][0].rmse < 1e-5
assert sisso.models[0][0].test_rmse < 1e-5
assert sisso.models[0][0].rmse < 1e-4
assert sisso.models[0][0].test_rmse < 1e-4
if __name__ == "__main__":
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment