diff --git a/tests/read_layers_regr.npz b/tests/read_layers_regr.npz index 4741101..4302541 100644 Binary files a/tests/read_layers_regr.npz and b/tests/read_layers_regr.npz differ diff --git a/tests/test_read_layers.py b/tests/test_read_layers.py index c2cc751..0dab2d4 100644 --- a/tests/test_read_layers.py +++ b/tests/test_read_layers.py @@ -34,11 +34,11 @@ def write_layers_to_dir(dir_path): # Sorts arrays without mixing datapoints # This is needed because we dont need to guarantee read order def sort_result(ar): - ar = ar[ar[:, 4].argsort()] - ar = ar[ar[:, 3].argsort()] - ar = ar[ar[:, 0].argsort()] - ar = ar[ar[:, 1].argsort()] - ar = ar[ar[:, 2].argsort()] + ar = ar[ar[:, 4].argsort(kind='stable')] + ar = ar[ar[:, 3].argsort(kind='stable')] + ar = ar[ar[:, 0].argsort(kind='stable')] + ar = ar[ar[:, 1].argsort(kind='stable')] + ar = ar[ar[:, 2].argsort(kind='stable')] return ar