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
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):