1 import unittest 2 3 from test import test_support as support 4 5 # For scope testing. 6 g = "Global variable" 7 8 9 class DictComprehensionTest(unittest.TestCase): 10 11 def test_basics(self): 12 expected = {0: 10, 1: 11, 2: 12, 3: 13, 4: 14, 5: 15, 6: 16, 7: 17, 13 8: 18, 9: 19} 14 actual = {k: k + 10 for k in range(10)} 15 self.assertEqual(actual, expected) 16 17 expected = {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9} 18 actual = {k: v for k in range(10) for v in range(10) if k == v} 19 self.assertEqual(actual, expected) 20 21 def test_scope_isolation(self): 22 k = "Local Variable" 23 24 expected = {0: None, 1: None, 2: None, 3: None, 4: None, 5: None, 25 6: None, 7: None, 8: None, 9: None} 26 actual = {k: None for k in range(10)} 27 self.assertEqual(actual, expected) 28 self.assertEqual(k, "Local Variable") 29 30 expected = {9: 1, 18: 2, 19: 2, 27: 3, 28: 3, 29: 3, 36: 4, 37: 4, 31 38: 4, 39: 4, 45: 5, 46: 5, 47: 5, 48: 5, 49: 5, 54: 6, 32 55: 6, 56: 6, 57: 6, 58: 6, 59: 6, 63: 7, 64: 7, 65: 7, 33 66: 7, 67: 7, 68: 7, 69: 7, 72: 8, 73: 8, 74: 8, 75: 8, 34 76: 8, 77: 8, 78: 8, 79: 8, 81: 9, 82: 9, 83: 9, 84: 9, 35 85: 9, 86: 9, 87: 9, 88: 9, 89: 9} 36 actual = {k: v for v in range(10) for k in range(v * 9, v * 10)} 37 self.assertEqual(k, "Local Variable") 38 self.assertEqual(actual, expected) 39 40 def test_scope_isolation_from_global(self): 41 expected = {0: None, 1: None, 2: None, 3: None, 4: None, 5: None, 42 6: None, 7: None, 8: None, 9: None} 43 actual = {g: None for g in range(10)} 44 self.assertEqual(actual, expected) 45 self.assertEqual(g, "Global variable") 46 47 expected = {9: 1, 18: 2, 19: 2, 27: 3, 28: 3, 29: 3, 36: 4, 37: 4, 48 38: 4, 39: 4, 45: 5, 46: 5, 47: 5, 48: 5, 49: 5, 54: 6, 49 55: 6, 56: 6, 57: 6, 58: 6, 59: 6, 63: 7, 64: 7, 65: 7, 50 66: 7, 67: 7, 68: 7, 69: 7, 72: 8, 73: 8, 74: 8, 75: 8, 51 76: 8, 77: 8, 78: 8, 79: 8, 81: 9, 82: 9, 83: 9, 84: 9, 52 85: 9, 86: 9, 87: 9, 88: 9, 89: 9} 53 actual = {g: v for v in range(10) for g in range(v * 9, v * 10)} 54 self.assertEqual(g, "Global variable") 55 self.assertEqual(actual, expected) 56 57 def test_global_visibility(self): 58 expected = {0: 'Global variable', 1: 'Global variable', 59 2: 'Global variable', 3: 'Global variable', 60 4: 'Global variable', 5: 'Global variable', 61 6: 'Global variable', 7: 'Global variable', 62 8: 'Global variable', 9: 'Global variable'} 63 actual = {k: g for k in range(10)} 64 self.assertEqual(actual, expected) 65 66 def test_local_visibility(self): 67 v = "Local variable" 68 expected = {0: 'Local variable', 1: 'Local variable', 69 2: 'Local variable', 3: 'Local variable', 70 4: 'Local variable', 5: 'Local variable', 71 6: 'Local variable', 7: 'Local variable', 72 8: 'Local variable', 9: 'Local variable'} 73 actual = {k: v for k in range(10)} 74 self.assertEqual(actual, expected) 75 self.assertEqual(v, "Local variable") 76 77 def test_illegal_assignment(self): 78 with self.assertRaisesRegexp(SyntaxError, "can't assign"): 79 compile("{x: y for y, x in ((1, 2), (3, 4))} = 5", "<test>", 80 "exec") 81 82 with self.assertRaisesRegexp(SyntaxError, "can't assign"): 83 compile("{x: y for y, x in ((1, 2), (3, 4))} += 5", "<test>", 84 "exec") 85 86 87 def test_main(): 88 support.run_unittest(__name__) 89 90 if __name__ == "__main__": 91 test_main() 92
