Update f4jumble.py

Co-authored-by: Daira Hopwood <daira@jacaranda.org>
2021-05-12 13:06:52 +01:00 · 2021-05-12 13:06:52 +01:00 · 232aaa3474
parent bd1eef9c16
commit 232aaa3474
1 changed files with 3 additions and 1 deletions
--- a/f4jumble.py
+++ b/f4jumble.py
@ -99,7 +99,9 @@ def main():
        16448,
    ]:
        M = rand.b(l_M)
-        assert f4jumble_inv(f4jumble(M)) == M
+        jumbled = f4jumble(M)
+        assert len(jumbled) == len(M)
+        assert f4jumble_inv(jumbled) == M
        test_vectors.append(M)

    test_vectors = [{