Update and rename test_all.py to test_sim_data_scalar.py

XMU-Kuangnan-Fang-Team · Oct 26, 2024 · 64445e6 · 64445e6
1 parent ef12f0f
commit 64445e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 32 deletions.
diff --git a/pytest/test_all.py b/pytest/test_all.py
diff --git a/pytest/test_sim_data_scalar.py b/pytest/test_sim_data_scalar.py
@@ -0,0 +1,6 @@
+from GENetLib.sim_data_scalar import sim_data_scalar
+
+def test_sim_data_scalar():
+    scalar_survival_linear = sim_data_scalar(rho_G=0.25, rho_E=0.3, dim_G=500, dim_E=5, n=1500, dim_E_Sparse=2, ytype='Survival', n_inter=30)
+    assert scalar_survival_linear[0] is not None
+    assert scalar_survival_linear[1] is not None