Supporting Python 2.6

cool-RR · cool-RR · commit ae890c4a5021 · 2015-07-19T00:15:40.000+03:00
diff --git a/source_py2/test_python_toolbox/test_nifty_collections/test_ordered_dict/test_with_stdlib_ordered_dict.py b/source_py2/test_python_toolbox/test_nifty_collections/test_ordered_dict/test_with_stdlib_ordered_dict.py
@@ -8,10 +8,10 @@
 from python_toolbox import cute_testing
 
 from python_toolbox.nifty_collections.ordered_dict import OrderedDict
+from python_toolbox.nifty_collections.ordered_dict import StdlibOrderedDict
 
 
 def test():
-    from collections import OrderedDict as StdlibOrderedDict
 
     ordered_dict = OrderedDict(((1, 'a'), (2, 'b'), (3, 'c')))
     stdlib_ordered_dict = StdlibOrderedDict(((1, 'a'), (2, 'b'), (3, 'c')))
diff --git a/source_py2/test_python_toolbox/test_nifty_collections/test_weak_key_default_dict/test.py b/source_py2/test_python_toolbox/test_nifty_collections/test_weak_key_default_dict/test.py
@@ -40,9 +40,10 @@ def test():
     )
     
     assert set(wkd_dict.iterkeys()) == set(wkd_dict.keys()) == \
-           {ref() for ref in wkd_dict.iterkeyrefs()} == \
-           {ref() for ref in wkd_dict.keyrefs()} == \
-           {weakreffable_object_0, weakreffable_object_1, weakreffable_object_2}
+           set((ref() for ref in wkd_dict.iterkeyrefs())) == \
+           set((ref() for ref in wkd_dict.keyrefs())) == \
+           set((weakreffable_object_0, weakreffable_object_1,
+                weakreffable_object_2))
     
     weakreffable_object_3 = WeakreffableObject()
     wkd_dict[weakreffable_object_3] = 123
