From 34b9d9d931850426714703fdcd6b1fbb9730b8b7 Mon Sep 17 00:00:00 2001 From: Niko Aarnio Date: Wed, 11 Oct 2023 17:18:22 +0300 Subject: [PATCH] Update wofe test --- tests/prediction/weights_of_evidence_test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/prediction/weights_of_evidence_test.py b/tests/prediction/weights_of_evidence_test.py index 315c186f..de282040 100644 --- a/tests/prediction/weights_of_evidence_test.py +++ b/tests/prediction/weights_of_evidence_test.py @@ -18,9 +18,8 @@ def test_weights_of_evidence(): """Test that weights of evidence works as intended.""" - df, rasters, raster_meta = weights_of_evidence_calculate_weights(evidence_raster, deposits) + df, rasters, raster_meta, _, _ = weights_of_evidence_calculate_weights(evidence_raster, deposits) - print(df["Studentized contrast"]) np.testing.assert_equal(df.shape[1], 10) # 10 columns for unique weights np.testing.assert_equal(df.shape[0], 8) # 8 classes in the test data np.testing.assert_equal(len(rasters), 3) # 3 rasters should be generated with default rasters_to_generate