From 10f612815e0bbbc109240f1333d0ddb2ea879274 Mon Sep 17 00:00:00 2001
From: Barry57 <160387047+Barry57@users.noreply.github.com>
Date: Sun, 27 Oct 2024 18:11:37 +0800
Subject: [PATCH] Update test_func_ge.py

---
 pytest/test_func_ge.py | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py
index 8a4e71f..1cf0e08 100644
--- a/pytest/test_func_ge.py
+++ b/pytest/test_func_ge.py
@@ -5,8 +5,23 @@
 
 def test_func_ge():
   func_continuous = sim_data_func(n=500, 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=False)
-  assert func_ge_res is not None
+  func_ge_res_1 = 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=False)
+  func_ge_res_2 = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 
+                          'Binary', 'Fourier', 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=0, ratio=[7, 3], plot_res=True)
+  func_ge_res_3 = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 
+                          'Survival', 'Monomial', 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=0, ratio=[7, 3], plot_res=True, plot_beta = False) 
+  func_ge_res_4 = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 
+                          'Survival', 'Exponential', 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, t=0.01, Bsplines=5, norder1=4, 
+                          model=None, split_type=0, ratio=[7, 3], plot_res=True)  
+  assert func_ge_res_1 is not None
+  assert func_ge_res_2 is not None
+  assert func_ge_res_3 is not None
+  assert func_ge_res_4 is not None