From f34008a741e6171e43b023ccd16174dabd3d1a87 Mon Sep 17 00:00:00 2001 From: Igor Rodionov Date: Sat, 25 Jan 2025 11:56:22 +0100 Subject: [PATCH] Added reverse test --- tests/test_order.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_order.py b/tests/test_order.py index cbba4ad9..0496c06b 100644 --- a/tests/test_order.py +++ b/tests/test_order.py @@ -63,3 +63,12 @@ def test_with_return_alphabet(self): exists_array, exists_alphabet = order(X, True) assert_array_equal(expected_alphabet, exists_alphabet) assert_array_equal(expected_array, exists_array) + + def test_reverse(self): + length = np.random.randint(1, 50000) + alphabet = np.arange(0, np.fix(length * 0.2), dtype=int) + X = np.random.choice(alphabet, length) + + exists_order, exists_alphabet = order(X, True) + X_restore = exists_alphabet[exists_order] + assert_array_equal(X, X_restore)