Update scale_range test

constantinpape · constantinpape · commit 529bb56a4ed2 · 2021-11-11T21:06:49.000+01:00
diff --git a/tests/prediction_pipeline/test_preprocessing.py b/tests/prediction_pipeline/test_preprocessing.py
@@ -131,8 +131,9 @@ def test_scale_range():
         )
     )
 
+    eps = 1.0e-6
     mi, ma = np_data.min(), np_data.max()
-    exp_data = (np_data - mi) / (ma - mi)
+    exp_data = (np_data - mi) / (ma - mi + eps)
     expected = xr.DataArray(exp_data, dims=("x", "y"))
 
     result = preprocessing(data)
@@ -158,9 +159,10 @@ def test_scale_range_axes():
         )
     )
 
+    eps = 1.0e-6
     p_low = np.percentile(np_data, min_percentile, axis=(1, 2), keepdims=True)
     p_up = np.percentile(np_data, max_percentile, axis=(1, 2), keepdims=True)
-    exp_data = (np_data - p_low) / (p_up - p_low)
+    exp_data = (np_data - p_low) / (p_up - p_low + eps)
     expected = xr.DataArray(exp_data, dims=("c", "x", "y"))
 
     result = preprocessing(data)

Original file line number	Diff line number	Diff line change
`@@ -131,8 +131,9 @@ def test_scale_range():`
`131`	`131`	`)`
`132`	`132`	`)`
`133`	`133`
	`134`	`+ eps = 1.0e-6`
`134`	`135`	`mi, ma = np_data.min(), np_data.max()`
`135`		`- exp_data = (np_data - mi) / (ma - mi)`
	`136`	`+ exp_data = (np_data - mi) / (ma - mi + eps)`
`136`	`137`	`expected = xr.DataArray(exp_data, dims=("x", "y"))`
`137`	`138`
`138`	`139`	`result = preprocessing(data)`
`@@ -158,9 +159,10 @@ def test_scale_range_axes():`
`158`	`159`	`)`
`159`	`160`	`)`
`160`	`161`
	`162`	`+ eps = 1.0e-6`
`161`	`163`	`p_low = np.percentile(np_data, min_percentile, axis=(1, 2), keepdims=True)`
`162`	`164`	`p_up = np.percentile(np_data, max_percentile, axis=(1, 2), keepdims=True)`
`163`		`- exp_data = (np_data - p_low) / (p_up - p_low)`
	`165`	`+ exp_data = (np_data - p_low) / (p_up - p_low + eps)`
`164`	`166`	`expected = xr.DataArray(exp_data, dims=("c", "x", "y"))`
`165`	`167`
`166`	`168`	`result = preprocessing(data)`