Skip to content

Commit

Permalink
Minor fix for dask test
Browse files Browse the repository at this point in the history
  • Loading branch information
darribas authored Aug 18, 2023
1 parent 9f20b02 commit 6a9cc3f
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions tobler/tests/test_area_interpolators.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,12 +93,9 @@ def test_area_interpolate_categorical_dask():
.spatial_shuffle(by='hilbert', shuffle='tasks')
)
area = area_interpolate_dask(
source_df=sac1,
target_df=sac2,
extensive_variables=["TOT_POP"],
intensive_variables=["pct_poverty"],
source_dgdf=sac1,
target_dgdf=sac2,
categorical_variables=["animal"],
n_jobs=1,
)
assert_almost_equal(area.animal_cat.sum(), 32, decimal=0)
assert_almost_equal(area.animal_dog.sum(), 19, decimal=0)
Expand Down

0 comments on commit 6a9cc3f

Please sign in to comment.