Fix test_hash_array

tomwhite · tomwhite · commit df26b2b7acd3 · 2024-09-02T10:09:04.000+01:00
diff --git a/sgkit/tests/test_popgen.py b/sgkit/tests/test_popgen.py
@@ -712,6 +712,9 @@ def test_hash_array(n_rows, n_cols):
     _, expected_inverse, expected_counts = np.unique(
         x, axis=0, return_inverse=True, return_counts=True
     )
+    # following is needed due to https://github.com/numpy/numpy/issues/26738
+    # (workaround from https://github.com/lmcinnes/umap/issues/1138)
+    expected_inverse = expected_inverse.reshape(-1)
 
     # hash columns, then find unique column counts using the hash values
     h = hash_array(x)

Original file line number	Diff line number	Diff line change
`@@ -712,6 +712,9 @@ def test_hash_array(n_rows, n_cols):`
`712`	`712`	`_, expected_inverse, expected_counts = np.unique(`
`713`	`713`	`x, axis=0, return_inverse=True, return_counts=True`
`714`	`714`	`)`
	`715`	`+ # following is needed due to https://github.com/numpy/numpy/issues/26738`
	`716`	`+ # (workaround from https://github.com/lmcinnes/umap/issues/1138)`
	`717`	`+ expected_inverse = expected_inverse.reshape(-1)`
`715`	`718`
`716`	`719`	`# hash columns, then find unique column counts using the hash values`
`717`	`720`	`h = hash_array(x)`