From fbee5812d2d5f848f3be09b2ea72435365157c145549d6259634c914b6a2195b Mon Sep 17 00:00:00 2001 From: Sergey Matveev Date: Sat, 30 Nov 2024 19:14:27 +0300 Subject: [PATCH] Limit lists-strategy --- pyac/tests/test_list.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyac/tests/test_list.py b/pyac/tests/test_list.py index 6f3f969..9d386b6 100644 --- 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: -- 2.50.0