Update test_sim_data_scalar.py

Barry57 · Oct 28, 2024 · d511832 · d511832
1 parent d30af75
commit d511832
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytest/test_sim_data_scalar.py b/pytest/test_sim_data_scalar.py
@@ -2,6 +2,6 @@
 
 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
+    assert scalar_survival_linear['data'] is not None
+    assert scalar_survival_linear['interpos'] is not None