Commit 7aec32ff authored by Thomas Purcell's avatar Thomas Purcell
Browse files

Fix Python tests

Param tests only work if PARAMS is built
parent ed8541a2
......@@ -36,7 +36,6 @@ def test_abs_diff_node():
np.array(task_sizes_test, dtype=np.int32),
4,
2,
False,
)
data_1 = np.random.random(task_sizes_train[0]) * 1e10 + 1e-10
......
......@@ -24,7 +24,7 @@ def test_abs_node():
task_sizes_train = [90]
task_sizes_test = [10]
initialize_values_arr(task_sizes_train, task_sizes_test, 3, 2, False)
initialize_values_arr(task_sizes_train, task_sizes_test, 3, 2)
data_1 = np.random.random(task_sizes_train[0]) * 1e4 + 1e-10
test_data_1 = np.random.random(task_sizes_test[0]) * 1e4 + 1e-10
......
......@@ -37,7 +37,7 @@ def test_param_abs_diff_node():
task_sizes_train = [900]
task_sizes_test = [10]
initialize_values_arr(task_sizes_train, task_sizes_test, 2, 1)
initialize_values_arr(task_sizes_train, task_sizes_test, 2, 1, True)
data_1 = np.linspace(-20, 20, task_sizes_train[0])
test_data_1 = np.linspace(-19.99, 19.99, task_sizes_test[0])
......
......@@ -36,7 +36,7 @@ def test_param_add_node():
task_sizes_train = [90]
task_sizes_test = [10]
initialize_values_arr(task_sizes_train, task_sizes_test, 2, 1)
initialize_values_arr(np.array(task_sizes_train), np.array(task_sizes_test), 2, 1)
initialize_param_storage()
data_1 = np.linspace(-20, 20, task_sizes_train[0])
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment