diff options
Diffstat (limited to 'tests/test_extras_dictcursor.py')
| -rwxr-xr-x | tests/test_extras_dictcursor.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/tests/test_extras_dictcursor.py b/tests/test_extras_dictcursor.py index 119110e..c7f09d5 100755 --- a/tests/test_extras_dictcursor.py +++ b/tests/test_extras_dictcursor.py @@ -265,7 +265,6 @@ class ExtrasDictCursorRealTests(_DictCursorBase): self.assertEqual(r, r1) self.assertEqual(r['a'], r1['a']) self.assertEqual(r['b'], r1['b']) - self.assertEqual(r._column_mapping, r1._column_mapping) def testDictCursorRealWithNamedCursorFetchOne(self): self._testWithNamedCursorReal(lambda curs: curs.fetchone()) @@ -377,6 +376,22 @@ class ExtrasDictCursorRealTests(_DictCursorBase): self.assertEqual(r.pop('b', None), None) self.assertRaises(KeyError, r.pop, 'b') + def test_mod(self): + curs = self.conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor) + curs.execute("select 1 as a, 2 as b, 3 as c") + r = curs.fetchone() + r['d'] = 4 + self.assertEqual(list(r), ['a', 'b', 'c', 'd']) + self.assertEqual(list(r.keys()), ['a', 'b', 'c', 'd']) + self.assertEqual(list(r.values()), [1, 2, 3, 4]) + self.assertEqual(list( + r.items()), [('a', 1), ('b', 2), ('c', 3), ('d', 4)]) + + assert r['a'] == 1 + assert r['b'] == 2 + assert r['c'] == 3 + assert r['d'] == 4 + class NamedTupleCursorTest(ConnectingTestCase): def setUp(self): |