diff --git a/cubed/tests/test_gufunc.py b/cubed/tests/test_gufunc.py index f68e3b87..a06d2e65 100644 --- a/cubed/tests/test_gufunc.py +++ b/cubed/tests/test_gufunc.py @@ -22,7 +22,7 @@ def stats(x): r = np.random.normal(size=(10, 20, 30)) a = cubed.from_array(r, chunks=(5, 5, 30), spec=spec) actual = apply_gufunc(stats, "(i)->()", a, output_dtypes="f", vectorize=vectorize) - expected = np.mean(r, axis=-1) + expected = np.mean(r, axis=-1, dtype=np.float32) assert actual.compute().shape == expected.shape assert_allclose(actual.compute(), expected)