Update test_func_ge.py

Barry57 · Oct 26, 2024 · 294f341 · 294f341
1 parent 90a422d
commit 294f341
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py
@@ -1,7 +1,6 @@
 from GENetLib.sim_data_func import sim_data_func
 from GENetLib.func_ge import func_ge
 
-def test_func_ge():
-    func_continuous = sim_data_func(n=100, m=30, ytype='Continuous', seed=123)
-    func_ge_res = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 'Continuous', 'Bspline', num_hidden_layers=2, nodes_hidden_layer=[100,10], Learning_Rate2=0.035, L2=0.01, Learning_Rate1=0.02, L=0.01, Num_Epochs=50, nbasis1=5, params1=4, Bsplines=5, norder1=4, model=None, split_type=1, ratio=[3, 1, 1], plot_res=True)
-    assert func_ge_res is not None
+func_continuous = sim_data_func(n=100, m=30, ytype='Continuous', seed=123)
+func_ge_res = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 'Continuous', 'Bspline', num_hidden_layers=2, nodes_hidden_layer=[100,10], Learning_Rate2=0.035, L2=0.01, Learning_Rate1=0.02, L=0.01, Num_Epochs=50, nbasis1=5, params1=4, Bsplines=5, norder1=4, model=None, split_type=1, ratio=[3, 1, 1], plot_res=True)
+