fixing bug in tests

tfm000 · Nov 5, 2023 · cbed45d · cbed45d
1 parent 82100fb
commit cbed45d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sklarpy/tests/multivariate/test_fitted_dists.py b/sklarpy/tests/multivariate/test_fitted_dists.py
@@ -136,7 +136,7 @@ def test_fitted_plots(params_2d, params_3d, mvt_continuous_data):
         size=(mvt_continuous_data.shape[0], 3))
     kwargs: dict = {'num_points': 2, 'num_generate': 10, 'mc_num_generate': 10,
                     'show': False, 'show_progress': False}
-    for name in ('multivariate_normal', ):
+    for name in ('mvt_normal', ):
         _, fitted_2d, _ = get_dist(name, params_2d, mvt_continuous_data)
         _, fitted_3d, _ = get_dist(name, params_3d, mvt_continuous_data_3d)
         for func_str in ('marginal_pairplot', 'pdf_plot',

diff --git a/sklarpy/tests/multivariate/test_prefit_dists.py b/sklarpy/tests/multivariate/test_prefit_dists.py
@@ -269,7 +269,7 @@ def test_prefit_plots(params_2d, params_3d, mvt_continuous_data):
         size=(mvt_continuous_data.shape[0], 3))
     kwargs: dict = {'num_points': 2, 'num_generate': 10, 'mc_num_generate': 10,
                     'show': False, 'show_progress': False}
-    for name in ('multivariate_normal', ):
+    for name in ('mvt_normal', ):
         dist, _, dist_params_2d = get_dist(name, params_2d,
                                            mvt_continuous_data)
         _, _, dist_params_3d = get_dist(name, params_3d,