handling memory error in testing

Took 50 minutes
tfm000 · Oct 10, 2023 · 148570b · 148570b
1 parent 3a859d9
commit 148570b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/sklarpy/tests/multivariate/test_fitted_dists.py b/sklarpy/tests/multivariate/test_fitted_dists.py
@@ -33,8 +33,14 @@ def test_fitted_logpdf_pdf_cdf_mc_cdf(
                                            pd_mvt_discrete_data)
 
             for data in datasets:
-                output = func(x=data, match_datatype=True,
-                              num_generate=num_generate)
+                try:
+                    output = func(x=data, match_datatype=True,
+                                  num_generate=num_generate)
+                except MemoryError as e:
+                    print(f"MemoryError occured when calculating fitted "
+                          f"{name}.{func_str} \n{e}")
+                    continue
+
                 np_output = np.asarray(output)
                 n, d = np.asarray(data).shape
 

diff --git a/sklarpy/tests/multivariate/test_prefit_dists.py b/sklarpy/tests/multivariate/test_prefit_dists.py
@@ -161,8 +161,14 @@ def test_prefit_logpdf_pdf_cdf_mc_cdfs(
                                            pd_mvt_discrete_data)
 
             for data in datasets:
-                output = func(x=data, params=params, match_datatype=True,
-                              num_generate=num_generate)
+                try:
+                    output = func(x=data, params=params, match_datatype=True,
+                                  num_generate=num_generate)
+                except MemoryError as e:
+                    print(f"MemoryError occured when calculating pre-fit "
+                          f"{name}.{func_str} \n{e}")
+                    continue
+
                 np_output = np.asarray(output)
                 n, d = np.asarray(data).shape