diff --git a/pytest/test_sim_data_func.py b/pytest/test_sim_data_func.py new file mode 100644 index 0000000..8145bd6 --- /dev/null +++ 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