|
| 1 | +"""Tests for various assignment forms""" |
| 2 | + |
| 3 | +import unittest |
| 4 | + |
| 5 | + |
| 6 | +class SimpleAssignTest(unittest.TestCase): |
| 7 | + |
| 8 | + def test_basic(self): |
| 9 | + x = 42 |
| 10 | + self.assertEqual(x, 42) |
| 11 | + |
| 12 | + def test_multiple_targets(self): |
| 13 | + a = b = c = 10 |
| 14 | + self.assertEqual(a, 10) |
| 15 | + self.assertEqual(b, 10) |
| 16 | + self.assertEqual(c, 10) |
| 17 | + |
| 18 | + def test_tuple_assign(self): |
| 19 | + a, b = 1, 2 |
| 20 | + self.assertEqual(a, 1) |
| 21 | + self.assertEqual(b, 2) |
| 22 | + |
| 23 | + def test_list_assign(self): |
| 24 | + [a, b, c] = [4, 5, 6] |
| 25 | + self.assertEqual((a, b, c), (4, 5, 6)) |
| 26 | + |
| 27 | + def test_swap(self): |
| 28 | + a, b = 1, 2 |
| 29 | + a, b = b, a |
| 30 | + self.assertEqual(a, 2) |
| 31 | + self.assertEqual(b, 1) |
| 32 | + |
| 33 | + def test_chained(self): |
| 34 | + x = y = z = [] |
| 35 | + x.append(1) |
| 36 | + self.assertEqual(y, [1]) |
| 37 | + self.assertEqual(z, [1]) |
| 38 | + self.assertIs(x, y) |
| 39 | + |
| 40 | + |
| 41 | +class AugmentedAssignTest(unittest.TestCase): |
| 42 | + |
| 43 | + def test_iadd(self): |
| 44 | + x = 10 |
| 45 | + x += 5 |
| 46 | + self.assertEqual(x, 15) |
| 47 | + |
| 48 | + def test_isub(self): |
| 49 | + x = 10 |
| 50 | + x -= 3 |
| 51 | + self.assertEqual(x, 7) |
| 52 | + |
| 53 | + def test_imul(self): |
| 54 | + x = 4 |
| 55 | + x *= 3 |
| 56 | + self.assertEqual(x, 12) |
| 57 | + |
| 58 | + def test_ifloordiv(self): |
| 59 | + x = 10 |
| 60 | + x //= 3 |
| 61 | + self.assertEqual(x, 3) |
| 62 | + |
| 63 | + def test_imod(self): |
| 64 | + x = 10 |
| 65 | + x %= 3 |
| 66 | + self.assertEqual(x, 1) |
| 67 | + |
| 68 | + def test_ipow(self): |
| 69 | + x = 2 |
| 70 | + x **= 10 |
| 71 | + self.assertEqual(x, 1024) |
| 72 | + |
| 73 | + def test_iand(self): |
| 74 | + x = 0xFF |
| 75 | + x &= 0x0F |
| 76 | + self.assertEqual(x, 0x0F) |
| 77 | + |
| 78 | + def test_ior(self): |
| 79 | + x = 0x0F |
| 80 | + x |= 0xF0 |
| 81 | + self.assertEqual(x, 0xFF) |
| 82 | + |
| 83 | + def test_ixor(self): |
| 84 | + x = 0xFF |
| 85 | + x ^= 0x0F |
| 86 | + self.assertEqual(x, 0xF0) |
| 87 | + |
| 88 | + def test_ilshift(self): |
| 89 | + x = 1 |
| 90 | + x <<= 10 |
| 91 | + self.assertEqual(x, 1024) |
| 92 | + |
| 93 | + def test_irshift(self): |
| 94 | + x = 1024 |
| 95 | + x >>= 10 |
| 96 | + self.assertEqual(x, 1) |
| 97 | + |
| 98 | + def test_iadd_list(self): |
| 99 | + x = [1, 2] |
| 100 | + y = x |
| 101 | + x += [3, 4] |
| 102 | + self.assertEqual(x, [1, 2, 3, 4]) |
| 103 | + self.assertIs(x, y) # list += modifies in place |
| 104 | + |
| 105 | + def test_iadd_string(self): |
| 106 | + x = "hello" |
| 107 | + x += " world" |
| 108 | + self.assertEqual(x, "hello world") |
| 109 | + |
| 110 | + def test_imul_list(self): |
| 111 | + x = [1, 2] |
| 112 | + x *= 3 |
| 113 | + self.assertEqual(x, [1, 2, 1, 2, 1, 2]) |
| 114 | + |
| 115 | + |
| 116 | +class SubscriptAssignTest(unittest.TestCase): |
| 117 | + |
| 118 | + def test_list_index(self): |
| 119 | + a = [0, 0, 0] |
| 120 | + a[0] = 1 |
| 121 | + a[2] = 3 |
| 122 | + self.assertEqual(a, [1, 0, 3]) |
| 123 | + |
| 124 | + def test_list_negative(self): |
| 125 | + a = [1, 2, 3] |
| 126 | + a[-1] = 99 |
| 127 | + self.assertEqual(a, [1, 2, 99]) |
| 128 | + |
| 129 | + def test_list_slice(self): |
| 130 | + a = [1, 2, 3, 4, 5] |
| 131 | + a[1:3] = [20, 30] |
| 132 | + self.assertEqual(a, [1, 20, 30, 4, 5]) |
| 133 | + |
| 134 | + def test_dict_assign(self): |
| 135 | + d = {} |
| 136 | + d['key'] = 'value' |
| 137 | + self.assertEqual(d['key'], 'value') |
| 138 | + |
| 139 | + def test_nested_assign(self): |
| 140 | + a = [[0, 0], [0, 0]] |
| 141 | + a[0][1] = 42 |
| 142 | + self.assertEqual(a[0][1], 42) |
| 143 | + |
| 144 | + |
| 145 | +class AttributeAssignTest(unittest.TestCase): |
| 146 | + |
| 147 | + def test_instance_attr(self): |
| 148 | + class C: |
| 149 | + pass |
| 150 | + obj = C() |
| 151 | + obj.x = 42 |
| 152 | + self.assertEqual(obj.x, 42) |
| 153 | + |
| 154 | + def test_overwrite(self): |
| 155 | + class C: |
| 156 | + pass |
| 157 | + obj = C() |
| 158 | + obj.x = 1 |
| 159 | + obj.x = 2 |
| 160 | + self.assertEqual(obj.x, 2) |
| 161 | + |
| 162 | + def test_multiple_attrs(self): |
| 163 | + class C: |
| 164 | + pass |
| 165 | + obj = C() |
| 166 | + obj.a = 1 |
| 167 | + obj.b = 2 |
| 168 | + obj.c = 3 |
| 169 | + self.assertEqual(obj.a + obj.b + obj.c, 6) |
| 170 | + |
| 171 | + |
| 172 | +if __name__ == "__main__": |
| 173 | + unittest.main() |
0 commit comments