Create test_sim_data_func.py

Barry57 · Oct 26, 2024 · 17390c9 · 17390c9
1 parent 6c6b7bc
commit 17390c9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pytest/test_sim_data_func.py b/pytest/test_sim_data_func.py
@@ -0,0 +1,8 @@
+from GENetLib.sim_data_func import sim_data_func
+
+def test_sim_data_func():
+    func_continuous = sim_data_func(n=1500, m=30, ytype='Continuous', seed=123)
+    assert func_continuous['X'] is not None 
+    assert func_continuous['y'] is not None
+    assert func_continuous['z'] is not None
+    assert func_continuous['location'] is not None