Limit lists-strategy

diff --git a/pyac/tests/test_list.py b/pyac/tests/test_list.py
index 6f3f969f493ac2eefac8652a64dbc81746e4e99d2b101ebdfa18d54e2340ee0d..9d386b60c5e28c039503cf291e6240bf4cc42e9e28aa96a930edd90031718aca 100644 (file)
--- a/pyac/tests/test_list.py
+++ b/pyac/tests/test_list.py
@@ -16,7 +16,7 @@ class TestList(TestCase):
         encoded = dumps([])
         self.assertEqual(encoded, b"\x08\x00")
 
-    @given(lists(any_st))
+    @given(lists(any_st, max_size=4))
     def test_list_encode_non_empty(self, test_list: List) -> None:
         encoded = dumps(test_list)
         self.assertEqual(
@@ -29,14 +29,14 @@ class TestList(TestCase):
         self.assertEqual(decoded, [])
         self.assertEqual(remaining, b"")
 
-    @given(lists(any_st), binary(max_size=20))
+    @given(lists(any_st, max_size=4), binary(max_size=20))
     def test_list_decode_non_empty(self, test_list: List, junk: bytes) -> None:
         encoded = b"\x08" + b"".join(dumps(i) for i in test_list) + b"\x00" + junk
         decoded, remaining = loads(encoded)
         self.assertEqual(decoded, test_list)
         self.assertEqual(remaining, junk)
 
-    @given(lists(any_st))
+    @given(lists(any_st, max_size=4))
     def test_no_eoc(self, test_list: List) -> None:
         encoded = dumps(test_list)[:-1]
         with self.assertRaises(NotEnoughData) as cm: