From: Sergey Matveev <stargrave@stargrave.org>
Date: Sat, 30 Nov 2024 16:29:02 +0000 (+0300)
Subject: Use more appropriate sets-strategy for map's keys
X-Git-Url: http://www.git.cypherpunks.su/?a=commitdiff_plain;h=9842045495876470086865629d985b28d851a0fb40d5683f6e0f9c7a6edf79a1;p=keks.git

Use more appropriate sets-strategy for map's keys

That way there won't be excess collapsing keys.
---

diff --git a/pyac/tests/test_map.py b/pyac/tests/test_map.py
index 89c41b4..caf2718 100644
--- a/pyac/tests/test_map.py
+++ b/pyac/tests/test_map.py
@@ -2,7 +2,7 @@ from unittest import TestCase
 
 from hypothesis import given
 from hypothesis.strategies import dictionaries
-from hypothesis.strategies import lists
+from hypothesis.strategies import sets
 
 from pyac import DecodeError
 from pyac import dumps
@@ -55,12 +55,12 @@ class TestMap(TestCase):
         self.assertEqual(decoded, test_map)
         self.assertSequenceEqual(tail, junk)
 
-    @given(lists(mapkey_st, max_size=4), junk_st)
+    @given(sets(mapkey_st, max_size=4), junk_st)
     def test_decode_to_set(self, keys, junk):
         test_map = {key: None for key in keys}
         encoded = dumps(test_map) + junk
         decoded, tail = loads(encoded, sets=True)
-        self.assertEqual(decoded, set(keys))
+        self.assertSetEqual(decoded, keys)
         self.assertSequenceEqual(tail, junk)
 
     def test_map_throws_when_decoding_unsorted_keys(self):