Skip to content
Snippets Groups Projects
Commit 5729b90a authored by Piotr Maślanka's avatar Piotr Maślanka
Browse files

switch over to list comprehensions in tests

parent f99d15cf
No related branches found
No related tags found
No related merge requests found
...@@ -16,16 +16,11 @@ class TestMiniJSON(unittest.TestCase): ...@@ -16,16 +16,11 @@ class TestMiniJSON(unittest.TestCase):
self.assertRaises(DecodingError, lambda: loads(b'\x82\x00')) self.assertRaises(DecodingError, lambda: loads(b'\x82\x00'))
def test_short_nonstring_key_dicts(self): def test_short_nonstring_key_dicts(self):
a = {} a = {i: i for i in range(20)}
for i in range(20):
a[i] = i
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
a = {} a = {i: i for i in range(300)}
for i in range(300):
a[i] = i
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
for i in range(70000): a = {i: i for i in range(66000)}
a[i] = i
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
def test_invalid_name_dict(self): def test_invalid_name_dict(self):
...@@ -74,25 +69,17 @@ class TestMiniJSON(unittest.TestCase): ...@@ -74,25 +69,17 @@ class TestMiniJSON(unittest.TestCase):
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
def test_long_dicts(self): def test_long_dicts(self):
a = {} a = {str(i): i for i in range(17)}
for i in range(17):
a[str(i)] = i
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
def test_dicts_not_string_keys(self): def test_dicts_not_string_keys(self):
a = {} a = {i: i for i in range(17)}
for i in range(17):
a[i] = i
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
def test_long_dicts_and_lists(self): def test_long_dicts_and_lists(self):
a = {} a = {str(i): i*2 for i in range(65535)}
for i in range(65535):
a[str(i)] = i*2
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
a = {} a = {str(i): i*2 for i in range(0x1FFFFF)}
for i in range(0x1FFFF):
a[str(i)] = i*2
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
a = list(range(0xFFFF)) a = list(range(0xFFFF))
self.assertSameAfterDumpsAndLoads(a) self.assertSameAfterDumpsAndLoads(a)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment