5656except NameError : # Python >= 3.0
5757 unicode = str
5858
59- try :
60- from collections import OrderedDict
61- except ImportError : # Python 2.6 or 3.0
62- OrderedDict = dict
59+ from collections import OrderedDict
6360
6461if str is bytes : # noinspection PyUnresolvedReferences
6562 from StringIO import StringIO
66- else :
63+ else : # Python >= 3.0
6764 from io import StringIO
6865
6966windows = os .name == 'nt'
@@ -226,11 +223,7 @@ def testAllDBAttributes(self):
226223 'unescape_bytea' , 'update' , 'upsert' ,
227224 'use_regtypes' , 'user' ,
228225 ]
229- # __dir__ is not called in Python 2.6 for old-style classes
230- db_attributes = dir (self .db ) if hasattr (
231- self .db .__class__ , '__class__' ) else self .db .__dir__ ()
232- db_attributes = [a for a in db_attributes
233- if not a .startswith ('_' )]
226+ db_attributes = [a for a in self .db .__dir__ () if not a .startswith ('_' )]
234227 self .assertEqual (attributes , db_attributes )
235228
236229 def testAttributeDb (self ):
@@ -1005,11 +998,6 @@ def testQueryFormatted(self):
1005998 # test with tuple, inline
1006999 q = f ("select %s, %s, %s, %s" , (3 , 2.5 , 'hello' , True ), inline = True )
10071000 r = q .getresult ()[0 ]
1008- if isinstance (r [1 ], Decimal ):
1009- # Python 2.6 cannot compare float and Decimal
1010- r = list (r )
1011- r [1 ] = float (r [1 ])
1012- r = tuple (r )
10131001 self .assertEqual (r , (3 , 2.5 , 'hello' , t ))
10141002 # test with dict
10151003 q = f ("select %(a)s::int, %(b)s::real, %(c)s::text, %(d)s::bool" ,
@@ -2944,8 +2932,7 @@ def testGetAsDict(self):
29442932 self .assertEqual (row .rgb , t [0 ])
29452933 self .assertEqual (row .name , t [1 ])
29462934 self .assertEqual (row ._asdict (), dict (rgb = t [0 ], name = t [1 ]))
2947- if OrderedDict is not dict : # Python > 2.6
2948- self .assertEqual (r .keys (), expected .keys ())
2935+ self .assertEqual (r .keys (), expected .keys ())
29492936 r = get_as_dict (table , keyname = 'rgb' )
29502937 self .assertIsInstance (r , OrderedDict )
29512938 expected = OrderedDict ((row [1 ], (row [0 ], row [2 ]))
@@ -2962,8 +2949,7 @@ def testGetAsDict(self):
29622949 self .assertEqual (row .id , t [0 ])
29632950 self .assertEqual (row .name , t [1 ])
29642951 self .assertEqual (row ._asdict (), dict (id = t [0 ], name = t [1 ]))
2965- if OrderedDict is not dict : # Python > 2.6
2966- self .assertEqual (r .keys (), expected .keys ())
2952+ self .assertEqual (r .keys (), expected .keys ())
29672953 r = get_as_dict (table , keyname = ['id' , 'rgb' ])
29682954 self .assertIsInstance (r , OrderedDict )
29692955 expected = OrderedDict ((row [:2 ], row [2 :]) for row in colors )
@@ -2983,8 +2969,7 @@ def testGetAsDict(self):
29832969 if named :
29842970 self .assertEqual (row .name , t [0 ])
29852971 self .assertEqual (row ._asdict (), dict (name = t [0 ]))
2986- if OrderedDict is not dict : # Python > 2.6
2987- self .assertEqual (r .keys (), expected .keys ())
2972+ self .assertEqual (r .keys (), expected .keys ())
29882973 r = get_as_dict (table , keyname = ['id' , 'rgb' ], scalar = True )
29892974 self .assertIsInstance (r , OrderedDict )
29902975 expected = OrderedDict ((row [:2 ], row [2 ]) for row in colors )
@@ -2995,8 +2980,7 @@ def testGetAsDict(self):
29952980 self .assertIsInstance (row , str )
29962981 t = expected [key ]
29972982 self .assertEqual (row , t )
2998- if OrderedDict is not dict : # Python > 2.6
2999- self .assertEqual (r .keys (), expected .keys ())
2983+ self .assertEqual (r .keys (), expected .keys ())
30002984 r = get_as_dict (table , keyname = 'rgb' , what = ['rgb' , 'name' ], scalar = True )
30012985 self .assertIsInstance (r , OrderedDict )
30022986 expected = OrderedDict ((row [1 ], row [2 ])
@@ -3008,8 +2992,7 @@ def testGetAsDict(self):
30082992 self .assertIsInstance (row , str )
30092993 t = expected [key ]
30102994 self .assertEqual (row , t )
3011- if OrderedDict is not dict : # Python > 2.6
3012- self .assertEqual (r .keys (), expected .keys ())
2995+ self .assertEqual (r .keys (), expected .keys ())
30132996 r = get_as_dict (table , what = 'id, name' ,
30142997 where = "rgb like '#b%'" , scalar = True )
30152998 self .assertIsInstance (r , OrderedDict )
@@ -3021,8 +3004,7 @@ def testGetAsDict(self):
30213004 self .assertIsInstance (row , str )
30223005 t = expected [key ]
30233006 self .assertEqual (row , t )
3024- if OrderedDict is not dict : # Python > 2.6
3025- self .assertEqual (r .keys (), expected .keys ())
3007+ self .assertEqual (r .keys (), expected .keys ())
30263008 expected = r
30273009 r = get_as_dict (table , what = ['name' , 'id' ],
30283010 where = ['id > 1' , 'id < 4' , "rgb like '#b%'" ,
@@ -3050,8 +3032,7 @@ def testGetAsDict(self):
30503032 r = get_as_dict (table , order = False )
30513033 self .assertIsInstance (r , dict )
30523034 self .assertEqual (r , expected )
3053- if dict is not OrderedDict : # Python > 2.6
3054- self .assertNotIsInstance (self , OrderedDict )
3035+ self .assertNotIsInstance (self , OrderedDict )
30553036 # test with arbitrary from clause
30563037 from_table = '(select id, lower(name) as n2 from "%s") as t2' % table
30573038 # primary key must be passed explicitly in this case
0 commit comments