update tests to support both 2023.0.1 and later versions

uxlfoundation · Mar 24, 2023 · cc22197 · cc22197
1 parent b0653fc
commit cc22197
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/daal4py/sklearn/ensemble/tests/test_decision_forest.py b/daal4py/sklearn/ensemble/tests/test_decision_forest.py
@@ -33,7 +33,7 @@
 
 ACCURACY_RATIO = 0.95 if daal_check_version((2021, 'P', 400)) else 0.85
 MSE_RATIO = 1.07
-if daal_check_version((2023, 'P', 100)):
+if daal_check_version((2023, 'P', 101)):
     LOG_LOSS_RATIO = 1.5
 elif daal_check_version((2021, 'P', 400)):
     LOG_LOSS_RATIO = 1.4

diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -186,9 +186,11 @@ def test_svm_batch(self):
     ('decision_forest_regression_hist_batch',
      'decision_forest_regression_batch.csv', lambda r: r[1].prediction, (2023, 'P', 1)),
     ('decision_forest_regression_default_dense_batch',
-     'decision_forest_regression_batch_20230101.csv', lambda r: r[1].prediction, (2023, 'P', 100)),
+     'decision_forest_regression_batch_20230101.csv', 
+     lambda r: r[1].prediction, (2023, 'P', 101)),
     ('decision_forest_regression_hist_batch',
-     'decision_forest_regression_batch_20230101.csv', lambda r: r[1].prediction, (2023, 'P', 100)),
+     'decision_forest_regression_batch_20230101.csv', 
+     lambda r: r[1].prediction, (2023, 'P', 101)),
     ('decision_tree_classification_batch', 'decision_tree_classification_batch.csv',
      lambda r: r[1].prediction),
     ('decision_tree_regression_batch', 'decision_tree_regression_batch.csv',