From 232aaa3474276ea4600f824fc8bf942a21a436b0 Mon Sep 17 00:00:00 2001
From: str4d <jack@z.cash>
Date: Wed, 12 May 2021 13:06:52 +0100
Subject: [PATCH] Update f4jumble.py

Co-authored-by: Daira Hopwood <daira@jacaranda.org>
---
 f4jumble.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/f4jumble.py b/f4jumble.py
index dee53bf..613d328 100644
--- 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 = [{