Skip to content

Commit

Permalink
Update wofe test
Browse files Browse the repository at this point in the history
  • Loading branch information
nmaarnio committed Oct 11, 2023
1 parent 47d15ff commit 34b9d9d
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/prediction/weights_of_evidence_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 34b9d9d

Please sign in to comment.