Fix test failure on 32bit architectures (#136)

avalentino · web-flow · commit b6c1e1a616cc · 2022-08-16T10:37:02.000-06:00
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -504,7 +504,7 @@ def _maybe_chunk(arr):
         fill_value=0,
         engine=engine,
     )
-    assert_equal(result, np.zeros(expected_shape, dtype=np.int64))
+    assert_equal(result, np.zeros(expected_shape, dtype=np.intp))
 
     # now when subsets are NaN
     # labels = np.array([0, 0, 1, 1, 1], dtype=float)

Original file line number	Diff line number	Diff line change
`@@ -504,7 +504,7 @@ def _maybe_chunk(arr):`
`504`	`504`	`fill_value=0,`
`505`	`505`	`engine=engine,`
`506`	`506`	`)`
`507`		`- assert_equal(result, np.zeros(expected_shape, dtype=np.int64))`
	`507`	`+ assert_equal(result, np.zeros(expected_shape, dtype=np.intp))`
`508`	`508`
`509`	`509`	`# now when subsets are NaN`
`510`	`510`	`# labels = np.array([0, 0, 1, 1, 1], dtype=float)`